Pārlūkot izejas kodu

Update download-model.py

oobabooga 3 gadi atpakaļ
vecāks
revīzija
1e541d4882
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      download-model.py

+ 1 - 1
download-model.py

@@ -42,7 +42,7 @@ def sanitize_branch_name(branch_name):
         raise ValueError("Invalid branch name. Only alphanumeric characters, period, underscore and dash are allowed.")
 
 if __name__ == '__main__':
-    model = args.model
+    model = args.MODEL
     if model[-1] == '/':
         model = model[:-1]
         branch = args.branch