소스 검색

Convert the download option (A-Z) to upper case

oobabooga 3 년 전
부모
커밋
bb1dac2f76
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      download-model.py

+ 1 - 1
download-model.py

@@ -71,7 +71,7 @@ def select_model_from_default_options():
 
     print()
     print("Input> ", end='')
-    choice = input()[0]
+    choice = input()[0].strip().upper()
     if choice == char:
         print("""\nThen type the name of your desired Hugging Face model in the format organization/name.