Update ppo_training.py

shibing624 · May 31, 2024 · f2cf523 · f2cf523
1 parent 04d217b
commit f2cf523
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppo_training.py b/ppo_training.py
@@ -36,7 +36,7 @@
 MODEL_CLASSES = {
     "bloom": (AutoConfig, BloomForCausalLM, BloomTokenizerFast),
     "chatglm": (AutoConfig, AutoModel, AutoTokenizer),
-    "llama": (AutoConfig, LlamaForCausalLM, LlamaTokenizer),
+    "llama": (AutoConfig, LlamaForCausalLM, AutoTokenizer),
     "baichuan": (AutoConfig, AutoModelForCausalLM, AutoTokenizer),
     "auto": (AutoConfig, AutoModelForCausalLM, AutoTokenizer),
 }