--- library_name: transformers tags: [] --- # Sample code ```python !pip install -U bitsandbytes !pip install -U transformers !pip install -U accelerate !pip install -U datasets !pip install -U peft !pip install ipywidgets --upgrade from transformers import ( AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, ) from peft import PeftModel import torch from tqdm import tqdm import json import re from google.colab import files HF_TOKEN = "YOUR TOKEN" your_path = '/elyza-tasks-100-TV_0.jsonl' model_id = "llm-jp/llm-jp-3-13b" adapter_id = "mss6/f4" model = PeftModel.from_pretrained(model, adapter_id, token = HF_TOKEN) datasets = [] with open(your_path, "r") as f: item = "" for line in f: line = line.strip() item += line if item.endswith("}"): datasets.append(json.loads(item)) item = "" from tqdm import tqdm results = [] for i in tqdm(range(100)): data = datasets[i] input = data["input"] prompt = f"""### 指示 {input} ### 回答 """ tokenized_input = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt").to(model.device) attention_mask = torch.ones_like(tokenized_input) with torch.no_grad(): outputs = model.generate( tokenized_input, attention_mask=attention_mask, max_new_tokens=1000, do_sample=False, repetition_penalty=1.2, pad_token_id=tokenizer.eos_token_id )[0] output = tokenizer.decode(outputs[tokenized_input.size(1):], skip_special_tokens=True) results.append({"task_id": data["task_id"], "input": input, "output": output}) jsonl_id = re.sub(".*/", "", 'ans') with open(f"/content/drive/MyDrive/{jsonl_id}-outputs.jsonl", 'w', encoding='utf-8') as f: for result in results: json.dump(result, f, ensure_ascii=False) f.write('\n') ``` ## Datasets ### Instruction tuning The models have been fine-tuned on the following datasets. | Language | Dataset | url | |:---|:---|:---| |Japanese| ichikara-instruction | [ichikara-instruction](https://liat-aip.sakura.ne.jp/wp/llm%E3%81%AE%E3%81%9F%E3%82%81%E3%81%AE%E6%97%A5%E6%9C%AC%E8%AA%9E%E3%82%A4%E3%83%B3%E3%82%B9%E3%83%88%E3%83%A9%E3%82%AF%E3%82%B7%E3%83%A7%E3%83%B3%E3%83%87%E3%83%BC%E3%82%BF%E4%BD%9C%E6%88%90/llm%E3%81%AE%E3%81%9F%E3%82%81%E3%81%AE%E6%97%A5%E6%9C%AC%E8%AA%9E%E3%82%A4%E3%83%B3%E3%82%B9%E3%83%88%E3%83%A9%E3%82%AF%E3%82%B7%E3%83%A7%E3%83%B3%E3%83%87%E3%83%BC%E3%82%BF-%E5%85%AC%E9%96%8B/) | | | Synthesized data from Elyza-tasks-100| [Elyza-tasks-100](https://huggingface.co/datasets/elyza/ELYZA-tasks-100)|