Update download-model.py
Browse files- download-model.py +4 -0
download-model.py
CHANGED
|
@@ -312,6 +312,10 @@ if __name__ == '__main__':
|
|
| 312 |
model = args.MODEL
|
| 313 |
specific_file = args.specific_file
|
| 314 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 315 |
if model is None:
|
| 316 |
print("Error: Please specify the model you'd like to download (e.g. 'python download-model.py facebook/opt-1.3b').")
|
| 317 |
sys.exit()
|
|
|
|
| 312 |
model = args.MODEL
|
| 313 |
specific_file = args.specific_file
|
| 314 |
|
| 315 |
+
print("download-model.py model: " + model)
|
| 316 |
+
print("download-model.py branch: " + branch)
|
| 317 |
+
print("download-model.py specific_file: " + specific_file)
|
| 318 |
+
|
| 319 |
if model is None:
|
| 320 |
print("Error: Please specify the model you'd like to download (e.g. 'python download-model.py facebook/opt-1.3b').")
|
| 321 |
sys.exit()
|