bezzam HF Staff commited on
Commit
8a0aa13
·
verified ·
1 Parent(s): 8073e81

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +0 -11
Dockerfile CHANGED
@@ -54,17 +54,6 @@ RUN pip install --no-cache-dir "git+https://github.com/Tencent-Hunyuan/AuK.git@$
54
  # ChatML turn; soundfile writes the generated wavs and the prompt clips stage 3 scores SIM on.
55
  RUN pip install --no-cache-dir "datasets<4.0" tqdm soundfile librosa
56
 
57
- # Bake the weights into the image so a job never spends its GPU time downloading ~20 GB:
58
- # * tencent/AuK 1.5B base (MIT, ungated) -> /opt/models/AuK
59
- # * tencent/AuK-Flash distilled 4-step (MIT, ungated) -> /opt/models/AuK-Flash
60
- # * Qwen/Qwen2.5-Omni-3B the text/audio encoder both checkpoints load at runtime
61
- # Each AuK release dir carries its own config.yaml + vae.safetensors, which AukInfer picks up
62
- # from next to the checkpoint. run_eval.py reads them via --checkpoint_dir / --qwen_dir and falls
63
- # back to a Hub snapshot if a dir is missing, so dropping any of these only costs download time.
64
- RUN python3 -c "from huggingface_hub import snapshot_download; snapshot_download('tencent/AuK', local_dir='/opt/models/AuK')"
65
- RUN python3 -c "from huggingface_hub import snapshot_download; snapshot_download('tencent/AuK-Flash', local_dir='/opt/models/AuK-Flash')"
66
- RUN python3 -c "from huggingface_hub import snapshot_download; snapshot_download('Qwen/Qwen2.5-Omni-3B', local_dir='/opt/models/Qwen2.5-Omni-3B')"
67
-
68
  # Copy the full repository
69
  COPY . /app
70
 
 
54
  # ChatML turn; soundfile writes the generated wavs and the prompt clips stage 3 scores SIM on.
55
  RUN pip install --no-cache-dir "datasets<4.0" tqdm soundfile librosa
56
 
 
 
 
 
 
 
 
 
 
 
 
57
  # Copy the full repository
58
  COPY . /app
59