Update download-model.py

This commit is contained in:
oobabooga 2023-01-21 00:43:00 -03:00 committed by GitHub
parent 18ef72d7c0
commit 1e541d4882
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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