Nekochu commited on
Commit
88ca206
·
1 Parent(s): a4a86a8

add einops + vector_quantize_pytorch for model loading

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -73,7 +73,8 @@ RUN curl -fL --retry 3 --retry-delay 5 -o /app/models/vae-BF16.gguf \
73
  RUN pip3 install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu \
74
  "gradio[mcp]==5.29.0" requests torch safetensors \
75
  transformers>=4.51.0 peft>=0.18.0 accelerate>=1.12.0 \
76
- loguru "torchaudio==2.4.0" "diffusers==0.30.3" lightning numpy tensorboard soundfile
 
77
 
78
  # Clone ACE-Step repo for training module
79
  RUN git clone --depth 1 https://github.com/ace-step/ACE-Step-1.5 /app/ace-step-source
 
73
  RUN pip3 install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu \
74
  "gradio[mcp]==5.29.0" requests torch safetensors \
75
  transformers>=4.51.0 peft>=0.18.0 accelerate>=1.12.0 \
76
+ loguru "torchaudio==2.4.0" "diffusers==0.30.3" lightning numpy tensorboard soundfile \
77
+ einops vector_quantize_pytorch
78
 
79
  # Clone ACE-Step repo for training module
80
  RUN git clone --depth 1 https://github.com/ace-step/ACE-Step-1.5 /app/ace-step-source