wip

backend-developers-ltd · Sep 5, 2024 · 2f78a1e · 2f78a1e
1 parent 0fd660f
commit 2f78a1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/compute_horde_prompt_gen/download_model.py b/src/compute_horde_prompt_gen/download_model.py
@@ -42,5 +42,5 @@
     )
     model.save_pretrained(args.model_path)
 
-    tokenizer = AutoTokenizer.from_pretrained(args.model_name)
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_PATHS[args.model_name])
     tokenizer.save_pretrained(args.model_path)