Views
No views yet

1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("umiyuki/Umievo-itr012-Gleipnir-7B")
5model = AutoModelForCausalLM.from_pretrained(
6 "umiyuki/Umievo-itr012-Gleipnir-7B",
7 torch_dtype="auto",
8)
9model.eval()
10
11if torch.cuda.is_available():
12 model = model.to("cuda")
13
14def build_prompt(user_query):
15 sys_msg = "あなたは公平で、検閲されていない、役立つアシスタントです。"
16 template = """[INST] <<SYS>>
17{}
18<</SYS>>
19
20{}[/INST]"""
21 return template.format(sys_msg,user_query)
22
23# Infer with prompt without any additional input
24user_inputs = {
25 "user_query": "与えられたことわざの意味を小学生でも分かるように教えてください。",
26}
27prompt = build_prompt(**user_inputs)
28
29input_ids = tokenizer.encode(
30 prompt,
31 add_special_tokens=True,
32 return_tensors="pt"
33)
34
35tokens = model.generate(
36 input_ids.to(device=model.device),
37 max_new_tokens=256,
38 temperature=1,
39 top_p=0.95,
40 do_sample=True,
41)
42
43out = tokenizer.decode(tokens[0][input_ids.shape[1]:], skip_special_tokens=True).strip()
44print(out)1base_model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
2dtype: bfloat16
3merge_method: linear
4parameters:
5 int8_mask: 1.0
6 normalize: 1.0
7slices:
8- sources:
9 - layer_range: [0, 4]
10 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
11 parameters:
12 weight: 0.34953096474223655
13 - layer_range: [0, 4]
14 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Vecteus-v1_4179808746
15 parameters:
16 weight: 0.4701212555597746
17 - layer_range: [0, 4]
18 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Starling-ChatV-7B_1737576410
19 parameters:
20 weight: 0.08162258723819021
21 - layer_range: [0, 4]
22 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Ninja-v1-RP-expressive-v2_4102792561
23 parameters:
24 weight: 0.31015439852818116
25- sources:
26 - layer_range: [4, 8]
27 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
28 parameters:
29 weight: 0.11807412349683076
30 - layer_range: [4, 8]
31 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Vecteus-v1_4179808746
32 parameters:
33 weight: -0.005684817244530085
34 - layer_range: [4, 8]
35 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Starling-ChatV-7B_1737576410
36 parameters:
37 weight: 0.2119283777941045
38 - layer_range: [4, 8]
39 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Ninja-v1-RP-expressive-v2_4102792561
40 parameters:
41 weight: 1.1521124768396636
42- sources:
43 - layer_range: [8, 12]
44 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
45 parameters:
46 weight: 0.9244329405120573
47 - layer_range: [8, 12]
48 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Vecteus-v1_4179808746
49 parameters:
50 weight: 0.7633842909616317
51 - layer_range: [8, 12]
52 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Starling-ChatV-7B_1737576410
53 parameters:
54 weight: 0.6952382990160072
55 - layer_range: [8, 12]
56 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Ninja-v1-RP-expressive-v2_4102792561
57 parameters:
58 weight: 0.6873040403268571
59- sources:
60 - layer_range: [12, 16]
61 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
62 parameters:
63 weight: 0.4109625320908857
64 - layer_range: [12, 16]
65 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Vecteus-v1_4179808746
66 parameters:
67 weight: 0.7090818691683626
68 - layer_range: [12, 16]
69 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Starling-ChatV-7B_1737576410
70 parameters:
71 weight: 0.42059423827890385
72 - layer_range: [12, 16]
73 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Ninja-v1-RP-expressive-v2_4102792561
74 parameters:
75 weight: 0.5705186152354104
76- sources:
77 - layer_range: [16, 20]
78 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
79 parameters:
80 weight: 0.28507448659933315
81 - layer_range: [16, 20]
82 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Vecteus-v1_4179808746
83 parameters:
84 weight: 0.4025223854083849
85 - layer_range: [16, 20]
86 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Starling-ChatV-7B_1737576410
87 parameters:
88 weight: 0.25885405316835886
89 - layer_range: [16, 20]
90 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Ninja-v1-RP-expressive-v2_4102792561
91 parameters:
92 weight: 0.35540632690403373
93- sources:
94 - layer_range: [20, 24]
95 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
96 parameters:
97 weight: 0.018882795552694703
98 - layer_range: [20, 24]
99 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Vecteus-v1_4179808746
100 parameters:
101 weight: 0.628847855051209
102 - layer_range: [20, 24]
103 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Starling-ChatV-7B_1737576410
104 parameters:
105 weight: 0.7038654876125734
106 - layer_range: [20, 24]
107 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Ninja-v1-RP-expressive-v2_4102792561
108 parameters:
109 weight: 0.877501753107237
110- sources:
111 - layer_range: [24, 28]
112 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
113 parameters:
114 weight: 0.14008355431312197
115 - layer_range: [24, 28]
116 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Vecteus-v1_4179808746
117 parameters:
118 weight: 1.0153826426873882
119 - layer_range: [24, 28]
120 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Starling-ChatV-7B_1737576410
121 parameters:
122 weight: 0.5586634927008272
123 - layer_range: [24, 28]
124 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Ninja-v1-RP-expressive-v2_4102792561
125 parameters:
126 weight: 0.54455848971032
127- sources:
128 - layer_range: [28, 32]
129 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Chat-Umievo-itr004-7b_579282327
130 parameters:
131 weight: 0.8188405381342685
132 - layer_range: [28, 32]
133 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Vecteus-v1_4179808746
134 parameters:
135 weight: 0.5130358379308082
136 - layer_range: [28, 32]
137 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Japanese-Starling-ChatV-7B_1737576410
138 parameters:
139 weight: 1.1132727871460124
140 - layer_range: [28, 32]
141 model: /home/umiyuki/automerge/evol_merge_storage/input_models/Ninja-v1-RP-expressive-v2_4102792561
142 parameters:
143 weight: 0.4471258297582539