dummy-foo
/

ChatGLM3-Japanese

Text Generation

feature-extraction

Model card Files Files and versions

dummy-foo commited on Jun 9, 2024

Commit

6f5ca5b

·

verified ·

1 Parent(s): 0aa9c28

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -21,8 +21,8 @@ pipeline_tag: text-generation
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained("dummy-foo/ChatGLM3-Japanese-Zero", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("dummy-foo/ChatGLM3-Japanese-Zero", trust_remote_code=True, device_map='cuda:0', torch_dtype=torch.float16)
 # test Chinese chat
 response, _ = model.chat(tokenizer, "你是谁", history=[])

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("dummy-foo/ChatGLM3-Japanese", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("dummy-foo/ChatGLM3-Japanese", trust_remote_code=True, device_map='cuda:0', torch_dtype=torch.float16)
 # test Chinese chat
 response, _ = model.chat(tokenizer, "你是谁", history=[])