add zerogpu compatible

app.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 import random
 from datetime import datetime
@@ -15,7 +16,7 @@ from diffusers import (
 from diffusers.utils import export_to_video
 from peft import PeftModel
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda"
 mcm_id = "yhzhai/mcm"
 basedir = os.getcwd()
 savedir = os.path.join(
@@ -152,6 +153,7 @@ cache_pipeline = {
 }
 
 
+@spaces.GPU
 def infer(
     base_model, variant, prompt, seed=0, randomize_seed=True, num_inference_steps=4
 ):

requirements.txt

@@ -1,5 +1,6 @@
 # --extra-index-url https://download.pytorch.org/whl/cu118
 # torch==2.1.2
+spaces
 torchvision==0.16.2
 git+https://github.com/yhZhai/diffusers.git
 transformers==4.36.2