diff --git a/10_download_model_122b.sh b/10_download_model_122b.sh index 03c130e..b2d3256 100755 --- a/10_download_model_122b.sh +++ b/10_download_model_122b.sh @@ -40,9 +40,14 @@ apptainer exec \ --env HF_HUB_CACHE="${HF_CACHE_DIR}/hub" \ --env XDG_CACHE_HOME="${HOME}/.cache" \ "$SIF_FILE" \ - huggingface-cli download "$MODEL_ID" \ - --local-dir "$TARGET_DIR" \ - --local-dir-use-symlinks False + python3 -c " +from huggingface_hub import snapshot_download +snapshot_download( + '${MODEL_ID}', + local_dir='${TARGET_DIR}', + local_dir_use_symlinks=False, +) +" echo "" echo "=== Download complete ==="