Spaces:

Omnibus
/

text-to-vid

Runtime error

Omnibus commited on Sep 3, 2023

Commit

8a31d41

1 Parent(s): c94e232

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from diffusers.utils import export_to_video
 device = "cuda" if torch.cuda.is_available() else "cpu"
 #model = gr.Interface.load("models/camenduru/text2_video_zero")
 # load pipeline
-pipe = DiffusionPipeline.from_pretrained("camenduru/text2_video_zero").to(device)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 #if device == "cuda":
     # optimize for GPU memory

 device = "cuda" if torch.cuda.is_available() else "cpu"
 #model = gr.Interface.load("models/camenduru/text2_video_zero")
 # load pipeline
+pipe = DiffusionPipeline.from_pretrained("camenduru/text2-video-zero").to(device)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 #if device == "cuda":
     # optimize for GPU memory