From 7b831a6776544413132397387a6ca5888b6921f5 Mon Sep 17 00:00:00 2001 From: x54-729 Date: Tue, 7 Nov 2023 20:23:14 +0800 Subject: [PATCH] typo --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 2461c99..866e33c 100644 --- a/README.md +++ b/README.md @@ -157,7 +157,7 @@ We conducted a comprehensive evaluation of InternLM using the open-source evalua To load the InternLM 7B Chat model using Transformers, use the following code: ```python -import +import torch from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("internlm/internlm-chat-7b-v1_1", trust_remote_code=True) # Set `torch_dtype=torch.float16` to load model in float16, otherwise it will be loaded as float32 and might cause OOM Error.