Torch 16

text_to_video.py

@@ -37,7 +37,7 @@ class TextToVideoTool(Tool):
             self.device = get_default_device()
 
         self.pipeline = DiffusionPipeline.from_pretrained(
-            self.default_checkpoint, torch_dtype=torch.float16, variant="fp16"
+            self.default_checkpoint, variant="fp16"
         )
         self.pipeline.to(self.device)