Update app.py

app.py

@@ -158,7 +158,7 @@ preset_options = [
     {"label": "896x608, 73 frames", "width": 896, "height": 608, "num_frames": 73},
     {"label": "896x544, 81 frames", "width": 896, "height": 544, "num_frames": 81},
     {"label": "832x544, 89 frames", "width": 832, "height": 544, "num_frames": 89},
-    {"label": "768x768, 200 frames", "width": 768, "height": 768, "num_frames": 160},
+    {"label": "768x768, 200 frames", "width": 768, "height": 768, "num_frames": 140},
     {"label": "768x512, 97 frames", "width": 768, "height": 512, "num_frames": 97},
     {"label": "800x480, 105 frames", "width": 800, "height": 480, "num_frames": 105},
     {"label": "736x480, 113 frames", "width": 736, "height": 480, "num_frames": 113},
@@ -250,7 +250,7 @@ def generate_video_from_text(
         "media_items": None,
     }
 
-    generator = torch.Generator(device="cpu").manual_seed(seed)
+    generator = torch.Generator(device="cuda").manual_seed(seed)
 
     def gradio_progress_callback(self, step, timestep, kwargs):
         progress((step + 1) / num_inference_steps)
@@ -346,7 +346,7 @@ def generate_video_from_image(
         "media_items": media_items,
     }
 
-    generator = torch.Generator(device="cpu").manual_seed(seed)
+    generator = torch.Generator(device="cuda").manual_seed(seed)
 
     def gradio_progress_callback(self, step, timestep, kwargs):
         progress((step + 1) / num_inference_steps)