Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,10 +4,10 @@ from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
|
|
4 |
from diffusers.utils import export_to_video
|
5 |
|
6 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
7 |
-
model = gr.Interface.load("models/camenduru/text2_video_zero")
|
8 |
# load pipeline
|
9 |
-
|
10 |
-
|
11 |
#if device == "cuda":
|
12 |
# optimize for GPU memory
|
13 |
# pipe.enable_model_cpu_offload()
|
|
|
4 |
from diffusers.utils import export_to_video
|
5 |
|
6 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
7 |
+
#model = gr.Interface.load("models/camenduru/text2_video_zero")
|
8 |
# load pipeline
|
9 |
+
pipe = DiffusionPipeline.from_pretrained("camenduru/text2_video_zero").to(device)
|
10 |
+
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
11 |
#if device == "cuda":
|
12 |
# optimize for GPU memory
|
13 |
# pipe.enable_model_cpu_offload()
|