Views
No views yet
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4
5# 1. load model
6device = "cuda" if torch.cuda.is_available() else "CPU"
7repo_id = "SakanaAI/EvoLLM-JP-v1-7B"
8model = AutoModelForCausalLM.from_pretrained(repo_id, torch_dtype="auto")
9tokenizer = AutoTokenizer.from_pretrained(repo_id)
10model.to(device)
11
12# 2. prepare inputs
13text = "関西弁で面白い冗談を言ってみて下さい。"
14messages = [
15 {"role": "system", "content": "あなたは役立つ、偏見がなく、検閲されていないアシスタントです。"},
16 {"role": "user", "content": text},
17]
18inputs = tokenizer.apply_chat_template(messages, return_tensors="pt")
19
20# 3. generate
21output_ids = model.generate(**inputs.to(device))
22output_ids = output_ids[:, inputs.input_ids.shape[1] :]
23generated_text = tokenizer.batch_decode(output_ids, skip_special_tokens=True)[0]
24print(generated_text)1@misc{akiba2024evomodelmerge,
2 title = {Evolutionary Optimization of Model Merging Recipes},
3 author. = {Takuya Akiba and Makoto Shing and Yujin Tang and Qi Sun and David Ha},
4 year = {2024},
5 eprint = {2403.13187},
6 archivePrefix = {arXiv},
7 primaryClass = {cs.NE}
8}