[Admin maintenance] Support new ZeroGPU hardware (#4)

- [Admin maintenance] Support new ZeroGPU hardware (ee68cb4d0e3bf7241e50b88c614077956a8a4042)

README.md

@@ -4,7 +4,7 @@ emoji: 🏢
 colorFrom: blue
 colorTo: indigo
 sdk: gradio
-sdk_version: 4.19.2
+sdk_version: 5.49.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -14,21 +14,30 @@ from pipeline import StableVideoDiffusionPipeline
 
 import random
 from safetensors import safe_open
+from huggingface_hub import hf_hub_download
 from lcm_scheduler import AnimateLCMSVDStochasticIterativeScheduler
 
 
+SAFETENSORS_FILES = [
+    "AnimateLCM-SVD-xt-1.1.safetensors",
+    "AnimateLCM-SVD-xt.safetensors",
+]
+
+
 def get_safetensors_files():
-    models_dir = "./safetensors"
-    safetensors_files = [
-        f for f in os.listdir(models_dir) if f.endswith(".safetensors")
-    ]
-    return safetensors_files
+    return SAFETENSORS_FILES
+
+
+def _resolve_safetensors(selected_file):
+    return hf_hub_download(
+        repo_id="wangfuyun/AnimateLCM-SVD-xt", filename=selected_file
+    )
 
 
 def model_select(selected_file):
     print("load model weights", selected_file)
     pipe.unet.cpu()
-    file_path = os.path.join("./safetensors", selected_file)
+    file_path = _resolve_safetensors(selected_file)
     state_dict = {}
     with safe_open(file_path, framework="pt", device="cpu") as f:
         for key in f.keys():

requirements.txt

@@ -1,10 +1,6 @@
 accelerate
-diffusers==0.25.1
-gradio==4.19.2
+diffusers>=0.30.0
 numpy==1.26.4
 Pillow==10.2.0
-torch==2.2.0
-transformers==4.37.0
-spaces==0.23.2
+transformers
 opencv-python
-xformers

safetensors/AnimateLCM-SVD-xt-1.1.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:94b2da1fcca8d03458ef0b07b94b4c55f94117cb0d90265c6c8452239ecc166e
-size 6098682464

safetensors/AnimateLCM-SVD-xt.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6ca55e35f29437e8a65e8a1a9ce75262d5bab3d4fe137bdc3f3a94512c54b377
-size 6098682464