From 8038a8985e35a1213275c0a015a3575f89755839 Mon Sep 17 00:00:00 2001 From: x54-729 Date: Mon, 6 Nov 2023 18:34:34 +0800 Subject: [PATCH] fix model path in readme --- README-ja-JP.md | 2 +- README-zh-Hans.md | 2 +- README.md | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README-ja-JP.md b/README-ja-JP.md index 77736fc..ac1efba 100644 --- a/README-ja-JP.md +++ b/README-ja-JP.md @@ -108,7 +108,7 @@ Transformers を使用して InternLM 7B チャットモデルをロードする ```python from transformers import AutoModelForCausalLM, AutoTokenizer -model_path = "/mnt/petrelfs/share_data/xingshuhao/internlm-chat-7b/" +model_path = "internlm/internlm-chat-7b" model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True) tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True) diff --git a/README-zh-Hans.md b/README-zh-Hans.md index edb64df..764eeca 100644 --- a/README-zh-Hans.md +++ b/README-zh-Hans.md @@ -183,7 +183,7 @@ InternLM-7B 包含了一个拥有70亿参数的基础模型和一个为实际场 ```python from transformers import AutoModelForCausalLM, AutoTokenizer -model_path = "/mnt/petrelfs/share_data/xingshuhao/internlm-chat-7b/" +model_path = "internlm/internlm-chat-7b" model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True) tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True) diff --git a/README.md b/README.md index 9983337..eb21166 100644 --- a/README.md +++ b/README.md @@ -180,7 +180,7 @@ The responses can be streamed using `stream_chat`: ```python from transformers import AutoModelForCausalLM, AutoTokenizer -model_path = "/mnt/petrelfs/share_data/xingshuhao/internlm-chat-7b/" +model_path = "internlm/internlm-chat-7b" model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True) tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)