Update 03-Baichuan2-7B-chat接入LangChain框架.md

zhouzhubin · Dec 18, 2023 · 775b53b · 775b53b
1 parent 63cb801
commit 775b53b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/BaiChuan/03-Baichuan2-7B-chat接入LangChain框架.md b/BaiChuan/03-Baichuan2-7B-chat接入LangChain框架.md
@@ -52,12 +52,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 import torch
 
 class baichuan2_LLM(LLM):
-    # 基于本地 InternLM 自定义 LLM 类
+    # 基于本地 Baichuan 自定义 LLM 类
     tokenizer : AutoTokenizer = None
     model: AutoModelForCausalLM = None
 
     def __init__(self, model_path :str):
-        # model_path: InternLM 模型路径
+        # model_path: Baichuan-7B-chat模型路径
         # 从本地初始化模型
         super().__init__()
         print("正在从本地加载模型...")