Downloads · 30 days
3
6% of all-time downloads
yossy0125/llm-jp-3-13b-it_lora
llm-jp-3-13b-it_lora is a machine learning model from yossy0125. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for peft. The card lists the license as apache-2.0.
- Developed by: [More Information Needed] - Funded by [optional]: [More Information Needed] - Shared by [optional]: [More Information Needed] - Model type: [More Information Needed] - Language(s) (NLP): [More Informat…
Downloads · 30 days
3
6% of all-time downloads
All-time downloads
53
Public
Repo size
501 MB
Likes
0
Public
Click a slice to open those files.
.safetensors501 MB · 99%
From the Hugging Face model README
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
Use the code below to get started with the model.
from transformers import AutoModelForCausalLM,AutoTokenizer,BitsAndBytesConfig
from peft import PeftModel,PeftConfig
import torch
HF_TOKEN = "your token"
model_name = "llm-jp/llm-jp-3-13b"
adapter_name = "yossy0125/llm-jp-3-13b-it_lora/"
#QLoRaの量子化に合わせる
bnb_config = BitsAndBytesConfig(
load_in_4bit=True,
bnb_4bit_quant_type= "nf4",
bnb_4bit_compute_dtype=torch.bfloat16,
)
#BaseModel
model = AutoModelForCausalLM.from_pretrained(
model_name,
quantization_config = bnb_config,
device_map="auto",
token=HF_TOKEN
)
tokenizer = AutoTokenizer.from_pretrained(model_name,trust_remote_code=True,token=HF_TOKEN)
#adapterをBaseModelに統合
model = PeftModel.from_pretrained(model,adapter_name,token=HF_TOKEN)
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)
outputs = None
with torch.no_grad():
outputs = model.generate(
tokenized_input,
attention_mask=attention_mask,
max_new_tokens=2048, #生成するトークン数
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)
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
BibTeX:
[More Information Needed]
APA:
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]
[More Information Needed]