Downloads · 30 days
0
m2af/OrionStarAI-Orion-14B-Base-adapter
OrionStarAI-Orion-14B-Base-adapter is a machine learning model from m2af. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Fine-tuned QLoRA Adapter for model OrionStarAI/Orion-14B-Base
Downloads · 30 days
0
Access
Public
Updated Mar 26, 2024
Repo size
133 MB
Likes
0
Public
Click a slice to open those files.
.safetensors131 MB · 99%
From the Hugging Face model README
Fine-tuned QLoRA Adapter for model OrionStarAI/Orion-14B-Base
Fine-tuned with Korean Sympathy Conversation dataset from AIHub
See more informations at our GitHub
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained(
"CurtisJeon/OrionStarAI-Orion-14B-Base-4bit",
low_cpu_mem_usage=True,
trust_remote_code=True,
device_map="auto",
)
model.config.use_cache = True
model.load_adapter("m2af/OrionStarAI-Orion-14B-Base-adapter", "loaded")
model.set_adapter("loaded")
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
tokenizer.pad_token = tokenizer.eos_token
tokenizer.padding_side = "right"
# Generate Sample
outputs = model.generate(**tokenizer("안녕하세요, 반갑습니다.", return_tensors="pt"))
print(outputs)