Views
No views yet

1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4tokenizer = AutoTokenizer.from_pretrained("stockmark/Stockmark-2-100B-Instruct-beta")
5model = AutoModelForCausalLM.from_pretrained(
6 "stockmark/Stockmark-2-100B-Instruct-beta", device_map="auto", torch_dtype=torch.bfloat16
7)
8
9instruction = "自然言語処理とは?"
10input_ids = tokenizer.apply_chat_template(
11 [{"role": "user", "content": instruction}], add_generation_prompt=True, return_tensors="pt"
12).to(model.device)
13
14with torch.inference_mode():
15 tokens = model.generate(
16 input_ids,
17 max_new_tokens = 512,
18 do_sample = True,
19 temperature = 0.7,
20 top_p = 0.95,
21 repetition_penalty = 1.05
22 )
23
24output = tokenizer.decode(tokens[0], skip_special_tokens=True)
25print(output)