Fix 122B model download to use Python API instead of huggingface-cli
Made-with: Cursor
This commit is contained in:
parent
eff76401ee
commit
a9ed1060cc
@ -40,9 +40,14 @@ apptainer exec \
|
|||||||
--env HF_HUB_CACHE="${HF_CACHE_DIR}/hub" \
|
--env HF_HUB_CACHE="${HF_CACHE_DIR}/hub" \
|
||||||
--env XDG_CACHE_HOME="${HOME}/.cache" \
|
--env XDG_CACHE_HOME="${HOME}/.cache" \
|
||||||
"$SIF_FILE" \
|
"$SIF_FILE" \
|
||||||
huggingface-cli download "$MODEL_ID" \
|
python3 -c "
|
||||||
--local-dir "$TARGET_DIR" \
|
from huggingface_hub import snapshot_download
|
||||||
--local-dir-use-symlinks False
|
snapshot_download(
|
||||||
|
'${MODEL_ID}',
|
||||||
|
local_dir='${TARGET_DIR}',
|
||||||
|
local_dir_use_symlinks=False,
|
||||||
|
)
|
||||||
|
"
|
||||||
|
|
||||||
echo ""
|
echo ""
|
||||||
echo "=== Download complete ==="
|
echo "=== Download complete ==="
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user