ssgyejin/KOTOX
Viewer β’ Updated β’ 13.8k β’ 88 β’ 2
How to use ssgyejin/llama-3-Korean-Bllossom-8B-deobfuscation-KOTOX with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("MLP-KTLim/llama-3-Korean-Bllossom-8B")
model = PeftModel.from_pretrained(base_model, "ssgyejin/llama-3-Korean-Bllossom-8B-deobfuscation-KOTOX")MLP-KTLim/llama-3-Korean-Bllossom-8B λ₯Ό νκ΅μ΄ ν μ€νΈ λλ ν 볡μ(deobfuscation) νμ€ν¬λ‘ νμΈνλν LoRA μ΄λν°μ λλ€. (μ 체 λͺ¨λΈμ΄ μλλΌ μ΄λν°μ΄λ―λ‘ λ² μ΄μ€ λͺ¨λΈμ μΉμ΄μ μ¬μ©ν©λλ€.)
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
base_model = "MLP-KTLim/llama-3-Korean-Bllossom-8B"
adapter = "ssgyejin/llama-3-Korean-Bllossom-8B-deobfuscation-KOTOX"
tokenizer = AutoTokenizer.from_pretrained(adapter, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
base_model, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True,
)
model = PeftModel.from_pretrained(model, adapter) # LoRA μ΄λν° κ²°ν©
model.eval()
messages = [{"role": "user", "content": "μ¬κΈ°μ λλ
νλ νκ΅μ΄ λ¬Έμ₯μ μ
λ ₯νμΈμ"}]
inputs = tokenizer.apply_chat_template(
messages, add_generation_prompt=True, return_tensors="pt"
).to(model.device)
out = model.generate(inputs, max_new_tokens=512, do_sample=False)
print(tokenizer.decode(out[0][inputs.shape[-1]:], skip_special_tokens=True))
μΆλ‘ μλκ° μ€μνλ©΄ model = model.merge_and_unload() λ‘ μ΄λν°λ₯Ό λ³ν©ν΄ μ¬μ©ν μ μμ΅λλ€.
MLP-KTLim/llama-3-Korean-Bllossom-8BBase model
meta-llama/Meta-Llama-3-8B