Downloads · 30 days
16
16% of all-time downloads
AIdenU/Mistral-7b-ko-Y24_v0.1
Mistral-7b-ko-Y24_v0.1 is a text generation model from AIdenU. Use it when you need the model to write or continue text. It is set up for transformers.
Downloads · 30 days
16
16% of all-time downloads
All-time downloads
103
Public
Repo size
29 GB
Likes
0
Public
Click a slice to open those files.
.bin14.5 GB · 100%
From the Hugging Face model README
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("AidenU/Mistral-7b-ko-Y24_v0.1", device_map="auto")
tokenizer = AutoTokenizer.from_pretrained("AidenU/Mistral-7b-ko-Y24_v0.1")
messages = [
{"role":"user", "content", "안녕하세요?"}
]
encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt")
inputs = encodeds.to("cuda")
model.to("cuda")
outputs = model.generated(
inputs,
max_new_tokens=256,
do_sample=True
)
decoded = tokenizer.batch_decode(outputs)
print(decoded[0])