Small refactor to tokenizer fix

2024-02-09 22:50:00 -05:00
parent b0e7b575f3
commit 7b3fd05542
1 changed files with 2 additions and 3 deletions
@@ -27,9 +27,8 @@ def move_files(src_dir, dest_dir):
        
 if __name__ == "__main__":
    model, download_dir = os.getenv("MODEL_NAME"), os.getenv("HF_HOME")
-    tokenizer = os.getenv("TOKENIZER_NAME", model)
-    if len(tokenizer) == 0:
-        tokenizer = model
+    tokenizer = os.getenv("TOKENIZER_NAME") or model
+
    revisions = {
        "model": os.getenv("MODEL_REVISION") or None,
        "tokenizer": os.getenv("TOKENIZER_REVISION") or None