Update app.py

app.py

@@ -8,10 +8,13 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # load pipeline
 pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16").to(device)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-
-# optimize for GPU memory
-pipe.enable_model_cpu_offload()
-pipe.enable_vae_slicing()
+if device == "cuda":
+    # optimize for GPU memory
+    pipe.enable_model_cpu_offload()
+    pipe.enable_vae_slicing()
+else:
+    pass
+    
 def ttv():
     
     # generate