Update download_model.py

This commit is contained in:
Justin Merrell
2023-11-21 20:10:37 -05:00
parent 5ca8e56bbe
commit cbde2cbf9c
+1 -1
View File
@@ -13,7 +13,7 @@ if __name__ == "__main__":
parser.add_argument('--model_name', type=str, default=MODEL_NAME)
parser.add_argument('--model_revision', type=str, default=MODEL_REVISION)
parser.add_argument('--model_base_path', type=str, default=MODEL_BASE_PATH)
parser.add_argument('--hugging_face_hub_token', type=str, default=HUGGING_FACE_HUB_TOKEN)
parser.add_argument('--hugging_face_hub_token', type=str)
args = parser.parse_args()