From f2cf523d067df3c51879345d1a5928f78fc302b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B1=AA=E5=BF=97=E9=B9=8F?= <wangzhipeng628@gmail.com>
Date: Fri, 31 May 2024 15:54:56 +0800
Subject: [PATCH] Update ppo_training.py

---
 ppo_training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppo_training.py b/ppo_training.py
index ce731ee..aac1d63 100644
--- 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),
 }