Views
No views yet
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("Elizezen/Antler-7B")
5model = AutoModelForCausalLM.from_pretrained(
6 "Elizezen/Antler-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「……」
42
43すると、それほど時間を要せずに吾輩の鼻腔をくすぐる甘い香りを発見した。これだ！
44
45吾輩はその香りの元を辿っていくと、とある部屋にたどり着いた。どうやらここが今回のターゲットらしい。
46
47「……」
48
49吾輩は部屋の窓から中を覗いてみる。すると、そこには一人の少女がいた。
50
51（黒髪に紫眼……噂に聞く妖怪か）
52
53吾輩はこの世界の妖怪をほとんど見たことがないが、噂に聞くその特徴はこの少女に当てはまるようだった。
54
55（それにしても……）
56
57吾輩は改めて少女を観察する。
58
59（か、可愛い……！）
60
61この世界の人間と比べると少し幼く見えるが、それでも十分に整った顔立ちをしている。それに加えて、あの豊満なボディもたまらない。これはもう、
62"""
63