CocoBro commited on
Commit
b312d65
·
1 Parent(s): 6b00c17
Files changed (2) hide show
  1. app.py +1 -6
  2. requirements.txt +4 -2
app.py CHANGED
@@ -166,10 +166,7 @@ def patch_paths_in_exp_config(exp_cfg: Dict[str, Any], repo_root: Path, qwen_roo
166
  exp_cfg["model"]["content_encoder"]["text_encoder"]["model_path"] = str(qwen_root)
167
 
168
 
169
- # -----------------------------
170
- # ZeroGPU 核心任务
171
- # -----------------------------
172
- # 学长说的就是这里:所有费资源的操作(加载+推理)都要放在这里面
173
  @spaces.GPU
174
  def run_edit(audio_file, caption, num_steps, guidance_scale, guidance_rescale, seed):
175
 
@@ -184,8 +181,6 @@ def run_edit(audio_file, caption, num_steps, guidance_scale, guidance_rescale, s
184
 
185
 
186
 
187
- torch.backends.cuda.matmul.allow_tf32 = False
188
- torch.backends.cudnn.allow_tf32 = False
189
 
190
  try:
191
  from utils.config import register_omegaconf_resolvers
 
166
  exp_cfg["model"]["content_encoder"]["text_encoder"]["model_path"] = str(qwen_root)
167
 
168
 
169
+
 
 
 
170
  @spaces.GPU
171
  def run_edit(audio_file, caption, num_steps, guidance_scale, guidance_rescale, seed):
172
 
 
181
 
182
 
183
 
 
 
184
 
185
  try:
186
  from utils.config import register_omegaconf_resolvers
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  gradio==4.26.0
2
- spaces>=0.30.0
3
  # --- Core Framework (Pinned Versions) ---
4
  torch==2.5.1
5
  torchvision==0.20.1
@@ -26,4 +26,6 @@ soundfile
26
  # --- Logging ---
27
  wandb
28
  tensorboard
29
- swanlab
 
 
 
1
  gradio==4.26.0
2
+
3
  # --- Core Framework (Pinned Versions) ---
4
  torch==2.5.1
5
  torchvision==0.20.1
 
26
  # --- Logging ---
27
  wandb
28
  tensorboard
29
+ swanlab
30
+
31
+ spaces>=0.30.0