Views
No views yet

1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("Elizezen/Hameln-japanese-mistral-7B")
5model = AutoModelForCausalLM.from_pretrained(
6 "Elizezen/Hameln-japanese-mistral-7B",
7 torch_dtype="auto",
8)
9model.eval()
10
11if torch.cuda.is_available():
12 model = model.to("cuda")
13
14input_ids = tokenizer.encode(
15 "むかしむかし、あるところに、おじいさんとおばあさんが住んでいました。 おじいさんは山へ柴刈りに、",
16 add_special_tokens=True,
17 return_tensors="pt"
18)
19
20tokens = model.generate(
21 input_ids.to(device=model.device),
22 max_new_tokens=512,
23 temperature=1,
24 top_p=0.95,
25 do_sample=True,
26)
27
28out = tokenizer.decode(tokens[0][input_ids.shape[1]:], skip_special_tokens=True).strip()
29print(out)
30
31"""
32output example:
33むかしむかし、あるところに、おじいさんとおばあさんが住んでいました。 おじいさんは山へ柴刈りに、おばあさんは田んぼの稲の手伝いをするなど、二人で力を合わせて楽しく暮らしていました。
34ある日のこと、その地方一帯に大きな台風がやって来ました。強風に飛ばされた木や、家屋などが次々と倒れる中、幸いにもおじいさんとおばあさんの住んでいた村は無事でした。
35しかし、近隣の小さな村では被害が出ていました。家屋は全壊、農作物は荒らされ、何より多くの命が失われていました。
36「可哀想に……」
37おばあさんは心を痛め、神様に祈りを捧げ続けました。
38「天上の神様!どうか、私達人間を守って下さい!」
39おばあさんの祈りが通じたのか、台風は急速に勢力を落とし、被害は最小限の内に治まりました。
40"""
41