[Admin maintenance] Support new ZeroGPU hardware

#13
by multimodalart HF Staff - opened
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +11 -1
  3. requirements.txt +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: ⚡
4
  colorFrom: blue
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 4.25.0
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
 
4
  colorFrom: blue
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 4.26.0
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
app.py CHANGED
@@ -166,7 +166,17 @@ torch.jit._state.disable()
166
  install_dependencies(enable_optimization=args.enable_optimization)
167
 
168
  # import after installation
169
- from opensora.datasets import IMG_FPS, save_sample
 
 
 
 
 
 
 
 
 
 
170
  from opensora.datasets.aspect import get_image_size, get_num_frames
171
  from opensora.models.text_encoder.t5 import text_preprocessing
172
  from opensora.utils.inference_utils import (
 
166
  install_dependencies(enable_optimization=args.enable_optimization)
167
 
168
  # import after installation
169
+ from opensora.datasets import save_sample
170
+
171
+ try:
172
+ from opensora.datasets import IMG_FPS
173
+ except ImportError:
174
+ # IMG_FPS was removed from opensora.datasets in the "Open-Sora 2.0" upstream
175
+ # rewrite; it was a constant (=120) meaning "this sample is a still image,
176
+ # not a video". requirements.txt now pins the git dep to the pre-rewrite
177
+ # commit so this except branch shouldn't normally trigger, but keep the
178
+ # fallback in case that pin needs to move forward again.
179
+ IMG_FPS = 120
180
  from opensora.datasets.aspect import get_image_size, get_num_frames
181
  from opensora.models.text_encoder.t5 import text_preprocessing
182
  from opensora.utils.inference_utils import (
requirements.txt CHANGED
@@ -1,3 +1,3 @@
1
  xformers
2
  transformers
3
- git+https://github.com/hpcaitech/Open-Sora.git
 
1
  xformers
2
  transformers
3
+ git+https://github.com/hpcaitech/Open-Sora.git@0028a6c9eb4bbb1678347b3cf3911d0f6eb475a5