kanana-safeguard-bnb-4bit
Collection
kanana-safeguard-bnb-4bit โข 3 items โข Updated
4bit ์์ํ๋ kakaocorp/kanana-safeguard-siren-8b ๋ชจ๋ธ์ ๋๋ค.
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
bnb_config = BitsAndBytesConfig(
load_in_4bit=True,
bnb_4bit_use_double_quant=True,
bnb_4bit_quant_type="nf4",
bnb_4bit_compute_dtype=torch.bfloat16
)
model = AutoModelForCausalLM.from_pretrained(
"nxtcloud-org/kanana-safeguard-siren-8b-4bit",
quantization_config=bnb_config,
device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained("kakaocrop/kanana-safeguard-siren-8b")
Apache License 2.0 (์๋ณธ ๋ชจ๋ธ๊ณผ ๋์ผ)
Base model
kakaocorp/kanana-safeguard-siren-8b