test
Browse files- app.py +1 -6
- 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 |
-
|
| 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
|