Merge branch 'main' into openai-sse-output

2024-02-21 19:56:55 -05:00
parent aed0408f19 2941db0fb8
commit 3549cf24d5
2 changed files with 3 additions and 2 deletions
@@ -27,7 +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)
+    tokenizer = os.getenv("TOKENIZER_NAME") or model
+
    revisions = {
        "model": os.getenv("MODEL_REVISION") or None,
        "tokenizer": os.getenv("TOKENIZER_REVISION") or None