Views
No views yet

1import transformers
2import torch
3
4if __name__ == "__main__":
5 model_name = "Replete-AI/Phi-Elothir"
6 tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
7
8 model = (
9 transformers.AutoModelForCausalLM.from_pretrained(
10 model_name,
11 )
12 .to("cuda:0")
13 .eval()
14 )
15
16 messages = [
17 {"role": "user", "content": "Hello, who are you?"}
18 ]
19 inputs = tokenizer.apply_chat_template(messages, return_tensors="pt").to(model.device)
20 input_ids_cutoff = inputs.size(dim=1)
21
22 with torch.no_grad():
23 generated_ids = model.generate(
24 input_ids=inputs,
25 use_cache=True,
26 max_new_tokens=512,
27 temperature=0.2,
28 top_p=0.95,
29 do_sample=True,
30 eos_token_id=tokenizer.eos_token_id,
31 pad_token_id=tokenizer.pad_token_id,
32 )
33
34 completion = tokenizer.decode(
35 generated_ids[0][input_ids_cutoff:],
36 skip_special_tokens=True,
37 )
38
39 print(completion)1dtype: float16
2merge_method: passthrough
3slices:
4 - sources:
5 - model: abacaj/phi-2-super
6 layer_range: [0,2]
7 - sources:
8 - model: abacaj/phi-2-super
9 layer_range: [1,3]
10 - sources:
11 - model: abacaj/phi-2-super
12 layer_range: [2,4]
13 - sources:
14 - model: abacaj/phi-2-super
15 layer_range: [3,5]
16 - sources:
17 - model: abacaj/phi-2-super
18 layer_range: [4,6]
19 - sources:
20 - model: abacaj/phi-2-super
21 layer_range: [5,7]
22 - sources:
23 - model: abacaj/phi-2-super
24 layer_range: [6,8]
25 - sources:
26 - model: abacaj/phi-2-super
27 layer_range: [7,9]
28 - sources:
29 - model: abacaj/phi-2-super
30 layer_range: [8,10]
31 - sources:
32 - model: abacaj/phi-2-super
33 layer_range: [9,11]
34 - sources:
35 - model: abacaj/phi-2-super
36 layer_range: [10,12]
37 - sources:
38 - model: abacaj/phi-2-super
39 layer_range: [11,13]
40 - sources:
41 - model: abacaj/phi-2-super
42 layer_range: [12,14]
43 - sources:
44 - model: abacaj/phi-2-super
45 layer_range: [13,15]
46 - sources:
47 - model: abacaj/phi-2-super
48 layer_range: [14,16]
49 - sources:
50 - model: abacaj/phi-2-super
51 layer_range: [15,17]
52 - sources:
53 - model: abacaj/phi-2-super
54 layer_range: [16,18]
55 - sources:
56 - model: abacaj/phi-2-super
57 layer_range: [17,19]
58 - sources:
59 - model: abacaj/phi-2-super
60 layer_range: [18,20]
61 - sources:
62 - model: abacaj/phi-2-super
63 layer_range: [19,21]
64 - sources:
65 - model: abacaj/phi-2-super
66 layer_range: [20,22]
67 - sources:
68 - model: abacaj/phi-2-super
69 layer_range: [21,23]
70 - sources:
71 - model: abacaj/phi-2-super
72 layer_range: [22,24]
73 - sources:
74 - model: abacaj/phi-2-super
75 layer_range: [23,25]
76 - sources:
77 - model: abacaj/phi-2-super
78 layer_range: [24,26]
79 - sources:
80 - model: abacaj/phi-2-super
81 layer_range: [25,27]
82 - sources:
83 - model: abacaj/phi-2-super
84 layer_range: [26,28]
85 - sources:
86 - model: abacaj/phi-2-super
87 layer_range: [27,29]
88 - sources:
89 - model: abacaj/phi-2-super
90 layer_range: [28,30]
91 - sources:
92 - model: abacaj/phi-2-super
93 layer_range: [29,31]
94 - sources:
95 - model: abacaj/phi-2-super
96 layer_range: [30,32]