SCUT-DLVCLab
/

TongGu-7B-Instruct

Model card Files Files and versions Community

JiahuanCao commited on Sep 28

Commit

46e9009

•

1 Parent(s): 57e5006

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -95,7 +95,7 @@ ACCN-INS数据集只能用于非商业研究目的。对于想要使用ACCN-INS
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_path = "DLVCLab/TongGu-7B-Instruct"
 model = AutoModelForCausalLM.from_pretrained(model_path, device_map='auto', torch_dtype=torch.bfloat16, trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_path = "SCUT-DLVCLab/TongGu-7B-Instruct"
 model = AutoModelForCausalLM.from_pretrained(model_path, device_map='auto', torch_dtype=torch.bfloat16, trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)