From b990a1b8ac75806c7428a3085d36a8fc3c401e7e Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Tue, 10 Feb 2026 14:31:43 -0800 Subject: [PATCH] Fixed #37 --- hf-download.sh | 23 +++++++---------------- 1 file changed, 7 insertions(+), 16 deletions(-) diff --git a/hf-download.sh b/hf-download.sh index 13af983..4c4b4c2 100755 --- a/hf-download.sh +++ b/hf-download.sh @@ -154,26 +154,17 @@ else fi # Convert to the directory pattern used by HuggingFace -# This is a heuristic - look for directories starting with the model name if [ -d "$HUB_PATH" ]; then - # Search for model directories matching the pattern - # Different quantizations of the same model will have different suffixes - # Try different patterns to find the downloaded model if [ -n "$ORG" ]; then - # Look for directories starting with "models--$ORG--$MODEL" - mapfile -t potential_dirs < <(find "$HUB_PATH" -maxdepth 1 -type d -name "models--${ORG}--${MODEL}*" 2>/dev/null) + MODEL_DIR="$HUB_PATH/models--${ORG}--${MODEL}" else - # Look for directories starting with the model name - mapfile -t potential_dirs < <(find "$HUB_PATH" -maxdepth 1 -type d -name "${MODEL}*" 2>/dev/null) - fi - - # If we found potential directories, use the most recently modified one - if [ ${#potential_dirs[@]} -gt 0 ]; then - # Sort by modification time, most recent first - IFS=$'\n' SORTED_DIRS=($(sort -r -k9,10 <<<"${potential_dirs[*]}")) - unset IFS - MODEL_DIR="${SORTED_DIRS[0]}" + # For models without org, check both patterns + if [ -d "$HUB_PATH/models--${MODEL}" ]; then + MODEL_DIR="$HUB_PATH/models--${MODEL}" + else + MODEL_DIR="$HUB_PATH/${MODEL}" + fi fi fi