Dockerfile: avoid repo models/ dependency; skip large ckpt; install yarngpt if missing

Dockerfile

@@ -65,16 +65,19 @@ RUN if [ -n "$HF_TOKEN" ]; then \
 # Note: Models will be downloaded lazily at runtime to save build storage
 # Pre-downloading is disabled to avoid storage limit issues
 
-# Download model assets at build time
+# Optional: download small config at build; skip large checkpoints
 RUN mkdir -p /code/models && \
-    wget -O /code/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml \
-    https://huggingface.co/novateur/WavTokenizer-medium-speech-75token/resolve/main/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml && \
-    cp /code/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml /code/models/ && \
-    wget --no-check-certificate 'https://drive.usercontent.google.com/download?id=1-ASeEkrn4HY49yZWHTASgfGFNXdVnLTt&export=download&confirm=t' -O /code/models/wavtokenizer_large_speech_320_24k.ckpt
+    wget -q -O /code/models/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml \
+    https://huggingface.co/novateur/WavTokenizer-medium-speech-75token/resolve/main/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml || true
 
 # Copy project files
 COPY . .
-RUN if [ -d "/code/yarngpt" ] && [ -f "/code/yarngpt/audiotokenizer.py" ]; then echo "Using vendored yarngpt from project"; else echo "ERROR: vendored yarngpt missing" && exit 1; fi
+# Install yarngpt if not vendored in repo
+RUN if [ -d "/code/yarngpt" ] && [ -f "/code/yarngpt/audiotokenizer.py" ]; then \
+      echo "Using vendored yarngpt from project"; \
+    else \
+      pip install --no-cache-dir git+https://github.com/saheedniyi02/yarngpt.git; \
+    fi
 
 # Expose FastAPI port
 EXPOSE 7860