Update video_diffusion/zero_shot/zero_shot_text2video.py

video_diffusion/zero_shot/zero_shot_text2video.py

@@ -13,9 +13,9 @@ class ZeroShotText2VideoGenerator:
 
     def load_model(self, model_id):
         if self.pipe is None:
-            self.pipe = TextToVideoZeroPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
-            self.pipe.to("cuda")
-            self.pipe.enable_xformers_memory_efficient_attention()
+            self.pipe = TextToVideoZeroPipeline.from_pretrained(model_id, torch_dtype=torch.float32).to("cpu")
+            self.pipe.to("cpu")
+            #self.pipe.enable_xformers_memory_efficient_attention()
             self.pipe.enable_attention_slicing()
 
         return self.pipe
@@ -162,3 +162,4 @@ class ZeroShotText2VideoGenerator:
                 ],
                 outputs=zero_shot_text2video_output,
             )
+