Views
No views yet
SOLAR-10.7B-Instruct-v1.0pip install transformers peft torch1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# Load base model and adapter
5base_model_name = "Upstage/SOLAR-10.7B-Instruct-v1.0"
6adapter_name = "Jay1121/solar-chatbot-final"
7
8tokenizer = AutoTokenizer.from_pretrained(base_model_name, use_fast=True)
9model = AutoModelForCausalLM.from_pretrained(
10 base_model_name,
11 torch_dtype="auto",
12 device_map="auto"
13)
14model = PeftModel.from_pretrained(model, adapter_name)
15
16# Generate response
17prompt = "안녕! 오늘 뭐 하지?" # "Hi! What should I do today?"
18inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
19outputs = model.generate(
20 **inputs,
21 max_new_tokens=256,
22 do_sample=True,
23 temperature=0.8,
24 pad_token_id=tokenizer.eos_token_id
25)
26response = tokenizer.decode(outputs[0], skip_special_tokens=True)
27print(response)1@misc{solar-chatbot-final,
2 author = {Jihee Cho},
3 title = {solar-chatbot-final: Korean Conversational LoRA Adapter for SOLAR-10.7B},
4 year = {2024},
5 publisher = {Hugging Face},
6 howpublished = {\url{https://huggingface.co/Jay1121/solar-chatbot-final}}
7}