Views
No views yet
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4tokenizer = AutoTokenizer.from_pretrained("dummy-foo/ChatGLM3-Japanese", trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained("dummy-foo/ChatGLM3-Japanese", trust_remote_code=True, device_map='cuda:0', torch_dtype=torch.float16)
6
7# test Chinese chat
8response, _ = model.chat(tokenizer, "你是谁", history=[])
9print("\nQ:你是谁,A:", response)
10response, _ = model.chat(tokenizer, "你喜欢甜粽子还是咸粽子", history=[])
11print("\nQ:你喜欢甜粽子还是咸粽子,A:", response)
12
13# test Japanese chat
14response, _ = model.chat(tokenizer, "あなたは誰ですか", history=[])
15print("\nQ:あなたは誰ですか,A:", response)
16response, _ = model.chat(tokenizer, "すみません、ちょっとお聞きしたいことがあるんですが", history=[])
17print("\nQ:すみません、ちょっとお聞きしたいことがあるんですが,A:", response)
18
19# test Japanese tokenizer
20print("\ntokens: ", tokenizer.tokenize("恥の多い生涯を送って来ました。自分には、人間の生活というものが、見当つかないのです。"))