detect device

app.py

@@ -29,9 +29,14 @@ def get_modelscope_pipeline(
     mcm_variant: Optional[str] = "WebVid",
 ):
     model_id = "ali-vilab/text-to-video-ms-1.7b"
-    pipe = DiffusionPipeline.from_pretrained(
-        model_id, torch_dtype=torch.float16, variant="fp16"
-    )
+    if torch.cuda.is_available():
+        pipe = DiffusionPipeline.from_pretrained(
+            model_id, torch_dtype=torch.float16, variant="fp16"
+        )
+    else:
+        pipe = DiffusionPipeline.from_pretrained(
+            model_id
+        )
     scheduler = LCMScheduler.from_pretrained(
         model_id,
         subfolder="scheduler",
@@ -82,14 +87,23 @@ def get_animatediff_pipeline(
     else:
         raise ValueError(f"Unknown real_variant {real_variant}")
 
-    adapter = MotionAdapter.from_pretrained(
-        motion_module_path, torch_dtype=torch.float16
-    )
-    pipe = AnimateDiffPipeline.from_pretrained(
-        model_id,
-        motion_adapter=adapter,
-        torch_dtype=torch.float16,
-    )
+    if torch.cuda.is_available():
+        adapter = MotionAdapter.from_pretrained(
+            motion_module_path, torch_dtype=torch.float16
+        )
+        pipe = AnimateDiffPipeline.from_pretrained(
+            model_id,
+            motion_adapter=adapter,
+            torch_dtype=torch.float16,
+        )
+    else:
+        adapter = MotionAdapter.from_pretrained(
+            motion_module_path
+        )
+        pipe = AnimateDiffPipeline.from_pretrained(
+            model_id,
+            motion_adapter=adapter,
+        )
     scheduler = LCMScheduler.from_pretrained(
         model_id,
         subfolder="scheduler",
@@ -306,6 +320,11 @@ with gr.Blocks(css=css) as demo:
         """
         )
 
+        gr.Markdown(
+            f"""
+        <p align="center"> Currently running on {device}.</p>
+        """
+        )
         with gr.Row():
             base_model = gr.Dropdown(
                 label="Base model",