.

options/Video_model/Model.py

@@ -49,10 +49,9 @@ def Video(
     video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
 
     if isinstance(image, Image.Image):
-        transform = transforms.ToTensor()
-        image = transform(image).unsqueeze(0).to(device).float()
+        image = transforms.ToTensor()(image).unsqueeze(0).to(device)
 
-    with torch.autocast(device):
+    with torch.autocast(device,dtype=torch.float32):
         frames = pipeline(
             image, height = height, width = width,
             num_inference_steps = num_inference_steps,