Fix model download

TheTerrasque · Mar 27, 2023 · 5fa8d45 · 5fa8d45
1 parent fdce6b1
commit 5fa8d45
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/run.sh b/docker/run.sh
@@ -6,9 +6,9 @@
 model="alpaca-native-4bit"
 
 if [ ! -e "/app/models/$model" ] ; then
-    echo "Downloading LLaMa model $model (metadata)"
+    echo "Downloading LLaMa model $model"
     if [ ! -e "/app/models/$model" ] ; then
-        python download-model.py --text-only ozcur/$model
+        python download-model.py ozcur/$model
     fi
 fi
 # if [ ! -e "/app/models/$model-4bit.pt" ] ; then