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

- [Admin maintenance] Support new ZeroGPU hardware (336d55d6d624108d984c8de53ec3f226fb7ae67d)

README.md

@@ -4,7 +4,7 @@ app_file: demo.py
 tags:
   - Text-to-Video
 sdk: gradio
-sdk_version: 4.37.2
+sdk_version: 6.20.0
 short_description: Text-to-Video
 emoji: 🤗
 colorFrom: blue

demo.py

@@ -2,6 +2,24 @@
 # imported by older gradio/oauth.py). Must run before `import spaces` (which imports gradio).
 import huggingface_hub
 import huggingface_hub.constants as _hub_const
+# newer huggingface_hub dropped the `use_auth_token` kwarg (renamed to `token`), still passed
+# by diffusers==0.24.0 (load_config, from_pretrained, download). Wrap the hub entry points it
+# uses so `use_auth_token` is translated to `token`.
+import functools as _functools
+def _drop_use_auth_token(fn):
+    @_functools.wraps(fn)
+    def _wrapped(*args, **kwargs):
+        if "use_auth_token" in kwargs:
+            _tok = kwargs.pop("use_auth_token")
+            kwargs.setdefault("token", _tok)
+        return fn(*args, **kwargs)
+    return _wrapped
+for _mod_fn in ("hf_hub_download", "snapshot_download", "model_info", "repo_info", "list_repo_files"):
+    if hasattr(huggingface_hub, _mod_fn):
+        setattr(huggingface_hub, _mod_fn, _drop_use_auth_token(getattr(huggingface_hub, _mod_fn)))
+for _api_fn in ("model_info", "repo_info", "snapshot_download", "hf_hub_download", "list_repo_files"):
+    if hasattr(huggingface_hub.HfApi, _api_fn):
+        setattr(huggingface_hub.HfApi, _api_fn, _drop_use_auth_token(getattr(huggingface_hub.HfApi, _api_fn)))
 # old diffusers/transformers import symbols removed from newer huggingface_hub; restore them
 if not hasattr(huggingface_hub, "cached_download"):
     huggingface_hub.cached_download = huggingface_hub.hf_hub_download
@@ -20,6 +38,11 @@ if not hasattr(huggingface_hub, "HfFolder"):
             try: huggingface_hub.logout()
             except Exception: pass
     huggingface_hub.HfFolder = HfFolder
+# transformers>=5 removed FLAX_WEIGHTS_NAME (still imported by diffusers==0.24.0
+# pipeline_utils); restore the constant so the old diffusers import succeeds.
+import transformers.utils as _tf_utils
+if not hasattr(_tf_utils, "FLAX_WEIGHTS_NAME"):
+    _tf_utils.FLAX_WEIGHTS_NAME = "flax_model.msgpack"
 import gradio as gr
 import os
 import spaces
@@ -223,17 +246,17 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(visible=True) as input_raws:
             with gr.Row():
-                with gr.Column(scale=1.0):
+                with gr.Column(scale=1):
                     text_input = gr.Textbox(show_label=True, interactive=True, label="Prompt")
 
             with gr.Row():
-                with gr.Column(scale=0.5):
+                with gr.Column(scale=1):
                     sample_method = gr.Dropdown(choices=["DDIM", "EulerDiscrete", "PNDM"], label="Sample Method", value="DDIM")
 
-                with gr.Column(scale=0.5):
+                with gr.Column(scale=1):
                     video_length = gr.Dropdown(choices=[1, 16], label="Video Length (1 for T2I and 16 for T2V)", value=16)
             with gr.Row():
-                with gr.Column(scale=1.0):
+                with gr.Column(scale=1):
                     scfg_scale = gr.Slider(
                         minimum=1,
                         maximum=50,
@@ -243,7 +266,7 @@ with gr.Blocks() as demo:
                         label="Guidence Scale",
                     )
             with gr.Row():
-                with gr.Column(scale=1.0):
+                with gr.Column(scale=1):
                     seed = gr.Slider(
                         minimum=1,
                         maximum=2147483647,
@@ -253,7 +276,7 @@ with gr.Blocks() as demo:
                         label="Seed",
                     )
             with gr.Row():
-                with gr.Column(scale=0.5):
+                with gr.Column(scale=1):
                     height = gr.Slider(
                         minimum=256,
                         maximum=768,
@@ -263,7 +286,7 @@ with gr.Blocks() as demo:
                         label="Height",
                     )
             # with gr.Row():
-                with gr.Column(scale=0.5):
+                with gr.Column(scale=1):
                     width = gr.Slider(
                         minimum=256,
                         maximum=768,
@@ -273,7 +296,7 @@ with gr.Blocks() as demo:
                         label="Width",
                     )
             with gr.Row():
-                with gr.Column(scale=1.0):
+                with gr.Column(scale=1):
                     diffusion_step = gr.Slider(
                         minimum=20,
                         maximum=250,
@@ -284,11 +307,11 @@ with gr.Blocks() as demo:
                     )
 
                          
-        with gr.Column(scale=0.6, visible=True) as video_upload:
+        with gr.Column(scale=1, visible=True) as video_upload:
             output = gr.Video(interactive=False, include_audio=True, elem_id="输出的视频") #.style(height=360)
 
             with gr.Row():
-                with gr.Column(scale=1.0, min_width=0):
+                with gr.Column(scale=1, min_width=0):
                     run = gr.Button(value="Generate", variant='primary')
 
     EXAMPLES = [

requirements.txt

@@ -1,6 +1,6 @@
-torch --index-url https://download.pytorch.org/whl/cu118
-torchvision --index-url https://download.pytorch.org/whl/cu118
-torchaudio --index-url https://download.pytorch.org/whl/cu118
+torch==2.8.0
+torchvision==0.23.0
+torchaudio==2.8.0
 timm
 diffusers[torch]==0.24.0
 accelerate