update

app.py

@@ -3,6 +3,7 @@ import torch
 import os
 import spaces
 import uuid
+import devicetorch
 
 from diffusers import AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler
 from diffusers.utils import export_to_video
@@ -26,11 +27,12 @@ step_loaded = None
 base_loaded = "Realistic"
 motion_loaded = None
 
-# Ensure model and scheduler are initialized in GPU-enabled function
-if not torch.cuda.is_available():
-    raise NotImplementedError("No GPU detected!")
-
-device = "cuda"
+## Ensure model and scheduler are initialized in GPU-enabled function
+#if not torch.cuda.is_available():
+#    raise NotImplementedError("No GPU detected!")
+#
+#device = "cuda"
+device = devicetorch.get(torch)
 dtype = torch.float16
 pipe = AnimateDiffPipeline.from_pretrained(bases[base_loaded], torch_dtype=dtype).to(device)
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
@@ -172,4 +174,4 @@ with gr.Blocks(css="style.css") as demo:
 
     gr.Markdown(MORE)
 
-demo.queue().launch()
+demo.queue().launch()

requirements.txt

@@ -1,7 +1,7 @@
 accelerate
 diffusers
 gradio
-torch
+#torch
 transformers
 opencv-python
-peft
+peft