Update video_diffusion/damo/damo_text2_video.py

video_diffusion/damo/damo_text2_video.py

@@ -14,11 +14,11 @@ class DamoText2VideoGenerator:
     def load_model(self, stable_model, scheduler):
         if self.pipe is None:
             self.pipe = DiffusionPipeline.from_pretrained(
-                stable_model, torch_dtype=torch.float16, variant="fp16"
+                stable_model, torch_dtype=torch.float32
             )
             self.pipe = get_scheduler_list(pipe=self.pipe, scheduler=scheduler)
-            self.pipe.to("cuda")
-            self.pipe.enable_xformers_memory_efficient_attention()
+            self.pipe.to("cpu")
+            #self.pipe.enable_xformers_memory_efficient_attention()
         return self.pipe
 
     def generate_video(
@@ -124,3 +124,4 @@ class DamoText2VideoGenerator:
             ],
             outputs=dano_output,
         )
+