Merge pull request #14 from raymondpau/fix-model-filename

fix: Fix failure to load successfully downloaded model
go-skynet · Aug 24, 2023 · b90804f · b90804f
2 parents 98061d6 + 40338c6
commit b90804f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/charts/local-ai/templates/deployment.yaml b/charts/local-ai/templates/deployment.yaml
@@ -58,7 +58,7 @@ spec:
                   auth=$(echo "$line" | awk '{print $2}')
 
                   if [ -n "$url" ]; then
-                      filename=$(basename "$url")
+                      filename=$(basename "$url" .bin)
 
                       if [ "$FORCE_DOWNLOAD" = false ] && [ -f "$MODEL_DIR/$filename" ]; then
                           echo "File $filename already exists. Skipping download."
@@ -70,9 +70,9 @@ spec:
                       echo "Downloading $filename"
 
                       if [ -n "$auth" ]; then
-                          wget -P "$MODEL_DIR" --header "Authorization: Basic $auth" "$url"
+                          wget --header "Authorization: Basic $auth" "$url" -O "$MODEL_DIR/$filename"
                       else
-                          wget -P "$MODEL_DIR" "$url"
+                          wget "$url" -O "$MODEL_DIR/$filename"
                       fi
 
                       if [ "$?" -ne 0 ]; then