update

app.py

@@ -13,6 +13,7 @@ import gradio as gr
 # from moviepy.editor import *
 from diffusers.utils.import_utils import is_xformers_available
 from diffusers.utils import load_image, export_to_video, export_to_gif
+import devicetorch
 
 import os
 import sys
@@ -25,6 +26,8 @@ from gradio_demo.utils_drag import *
 import warnings
 print("gr file", gr.__file__)
 
+DEVICE = devicetorch.get(torch)
+
 from huggingface_hub import hf_hub_download, snapshot_download
 
 os.makedirs("checkpoints", exist_ok=True)
@@ -705,7 +708,8 @@ if __name__ == "__main__":
                     3. Interpolate the images (according the path) with a click on "Run" button. <br>""")
         
         # device, args, height, width, model_length
-        Framer = Drag("cuda", args, 320, 512, 14)
+        #Framer = Drag("cuda", args, 320, 512, 14)
+        Framer = Drag(DEVICE, args, 320, 512, 14)
         first_frame_path = gr.State()
         last_frame_path = gr.State()
         tracking_points = gr.State([])

requirements.txt

@@ -1,14 +1,16 @@
-torch==2.0.0
-torchvision
+#torch==2.0.0
+#torchvision
 diffusers==0.24.0
 transformers==4.27.0
-xformers==0.0.18
+#xformers==0.0.18
 imageio==2.27.0
-decord==0.6.0
+#decord==0.6.0
 einops
 opencv-python
 av
 accelerate==0.27.2
 scipy
 colorlog
-numpy==1.24.3
+numpy==1.24.3
+devicetorch
+gradio