Update README.md

README.md

@@ -59,7 +59,7 @@ import sys
 from peft import PeftModel
 from transformers import AutoModel, AutoTokenizer
 sys.path.append('..')
-model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, device_map='auto')
+model = AutoModel.from_pretrained("THUDM/chatglm-6b", device_map='auto')
 model = PeftModel.from_pretrained(model, "model/chatglm_fitness_lora")#"model/chatglm_fitness_lora"为您下载本项目压缩包后，解压后本地lora目录
 model = model.half().cuda()  # fp16
 tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
@@ -125,7 +125,7 @@ import sys
 from peft import PeftModel
 from transformers import AutoModel, AutoTokenizer
 sys.path.append('..')
-model = AutoModel.from_pretrained("fb700/chatglm-fitness-RLHF", trust_remote_code=True, device_map='auto')#fb700/chatglm-fitness-RLHF为hg自动下载地址，如已经自行下载请替换
+model = AutoModel.from_pretrained("fb700/chatglm-fitness-RLHF",  device_map='auto')#fb700/chatglm-fitness-RLHF为hg自动下载地址，如已经自行下载请替换
 #model = PeftModel.from_pretrained(model, "model/chatglm_fitness_lora") # lora文件保存目录
 model = model.half().quantize(4).cuda()  # int4
 #model = model.half().quantize(8).cuda()  # int8