Views
No views yet
| Model Variants |
|---|
| Doujinshi-1.8b |
| Doujinshi-1.8b-instruct |
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3tokenizer = AutoTokenizer.from_pretrained("puwaer/Doujinshi-1.8b-instruct")
4model = AutoModelForCausalLM.from_pretrained("puwaer/Doujinshi-1.8b-instruct", device_map="auto", torch_dtype=torch.bfloat16)
5chat = [
6 {"role": "system", "content": "以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。"},
7 {"role": "user", "content": "avの良さを教えて"},
8]
9tokenized_input = tokenizer.apply_chat_template(chat, add_generation_prompt=True, tokenize=True, return_tensors="pt").to(model.device)
10with torch.no_grad():
11 output = model.generate(
12 tokenized_input,
13 max_new_tokens=100,
14 do_sample=True,
15 top_p=0.95,
16 temperature=0.7,
17 repetition_penalty=1.05,
18 )[0]
19print(tokenizer.decode(output))