Spaces:
Running
Running
fix: save_every_n_epochs=0, add demucs-infer to Dockerfile, debug adapter dir
Browse files- Dockerfile +1 -1
- app.py +7 -1
Dockerfile
CHANGED
|
@@ -78,7 +78,7 @@ RUN pip3 install --no-cache-dir --extra-index-url https://download.pytorch.org/w
|
|
| 78 |
"gradio[mcp]>=6.0.0,<7.0.0" requests torch safetensors \
|
| 79 |
"transformers>=4.51.0,<4.58.0" peft>=0.18.0 \
|
| 80 |
loguru "torchaudio==2.4.0" "diffusers==0.30.3" lightning numpy tensorboard soundfile \
|
| 81 |
-
einops vector_quantize_pytorch librosa mutagen
|
| 82 |
|
| 83 |
# Clone ACE-Step repo for training module
|
| 84 |
RUN git clone --depth 1 https://github.com/ace-step/ACE-Step-1.5 /app/ace-step-source
|
|
|
|
| 78 |
"gradio[mcp]>=6.0.0,<7.0.0" requests torch safetensors \
|
| 79 |
"transformers>=4.51.0,<4.58.0" peft>=0.18.0 \
|
| 80 |
loguru "torchaudio==2.4.0" "diffusers==0.30.3" lightning numpy tensorboard soundfile \
|
| 81 |
+
einops vector_quantize_pytorch librosa mutagen demucs-infer
|
| 82 |
|
| 83 |
# Clone ACE-Step repo for training module
|
| 84 |
RUN git clone --depth 1 https://github.com/ace-step/ACE-Step-1.5 /app/ace-step-source
|
app.py
CHANGED
|
@@ -730,7 +730,7 @@ def gradio_main():
|
|
| 730 |
warmup_steps=100,
|
| 731 |
weight_decay=0.01,
|
| 732 |
max_grad_norm=1.0,
|
| 733 |
-
save_every_n_epochs=
|
| 734 |
seed=42,
|
| 735 |
variant="turbo",
|
| 736 |
device="cpu",
|
|
@@ -769,6 +769,12 @@ def gradio_main():
|
|
| 769 |
_log("[OK] ace-server restarted successfully")
|
| 770 |
else:
|
| 771 |
_log("[WARN] ace-server may not have restarted -- check logs")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 772 |
adapter_safetensors = os.path.join(adapter_out, "adapter_model.safetensors")
|
| 773 |
if os.path.isfile(adapter_safetensors):
|
| 774 |
# Copy to a temp file so Gradio doesn't try to validate /app paths
|
|
|
|
| 730 |
warmup_steps=100,
|
| 731 |
weight_decay=0.01,
|
| 732 |
max_grad_norm=1.0,
|
| 733 |
+
save_every_n_epochs=0,
|
| 734 |
seed=42,
|
| 735 |
variant="turbo",
|
| 736 |
device="cpu",
|
|
|
|
| 769 |
_log("[OK] ace-server restarted successfully")
|
| 770 |
else:
|
| 771 |
_log("[WARN] ace-server may not have restarted -- check logs")
|
| 772 |
+
# Debug: list what's in the adapter directory
|
| 773 |
+
if os.path.isdir(adapter_out):
|
| 774 |
+
contents = os.listdir(adapter_out)
|
| 775 |
+
_log(f"[DEBUG] Adapter dir {adapter_out}: {contents}")
|
| 776 |
+
else:
|
| 777 |
+
_log(f"[DEBUG] Adapter dir {adapter_out} does NOT exist!")
|
| 778 |
adapter_safetensors = os.path.join(adapter_out, "adapter_model.safetensors")
|
| 779 |
if os.path.isfile(adapter_safetensors):
|
| 780 |
# Copy to a temp file so Gradio doesn't try to validate /app paths
|