Nekochu commited on
Commit
0e27e49
·
1 Parent(s): 829ed0c

fix: save_every_n_epochs=0, add demucs-infer to Dockerfile, debug adapter dir

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. 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=max(1, epochs // 2),
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