Use correct diffuser

livepeer · Apr 11, 2024 · b46a460 · b46a460
1 parent 76c0152
commit b46a460
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/runner/app/pipelines/text_to_video.py b/runner/app/pipelines/text_to_video.py
@@ -1,7 +1,7 @@
 from app.pipelines.base import Pipeline
 from app.pipelines.util import get_torch_device, get_model_dir
 
-from diffusers import StableVideoDiffusionPipeline
+from diffusers import DiffusionPipeline
 from huggingface_hub import file_download
 import torch
 import PIL
@@ -37,7 +37,7 @@ def __init__(self, model_id: str):
             kwargs["variant"] = "fp16"
 
         self.model_id = model_id
-        self.ldm = StableVideoDiffusionPipeline.from_pretrained(model_id, **kwargs)
+        self.ldm = DiffusionPipeline.from_pretrained(model_id, **kwargs)
         self.ldm.to(get_torch_device())
 
         if os.environ.get("SFAST"):