Downloads · 30 days
0
Kohsaku/gemma-2-9b-finetune-2
gemma-2-9b-finetune-2 is a machine learning model from Kohsaku. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as gemma.
- Developed by: Kohsaku - License: Gemma 2 License - Finetuned from model : google/gemma-2-9b
Downloads · 30 days
0
Access
Public
Updated Dec 9, 2024
Repo size
255 MB
Likes
0
Public
Click a slice to open those files.
.safetensors216 MB · 85%
From the Hugging Face model README
This gemma2 model was trained 2x faster with Unsloth and Huggingface's TRL library.
model_name = "Kohsaku/gemma-2-9b-finetune-2"
max_seq_length = 1024
dtype = None
load_in_4bit = True
model, tokenizer = FastLanguageModel.from_pretrained(
model_name = model_name,
max_seq_length = max_seq_length,
dtype = dtype,
load_in_4bit = load_in_4bit,
token = HF_TOKEN,
)
FastLanguageModel.for_inference(model)
text = "自然言語処理とは何か"
tokenized_input = tokenizer.encode(text, add_special_tokens=True , return_tensors="pt").to(model.device)
with torch.no_grad():
output = model.generate(
tokenized_input,
max_new_tokens = 1024,
use_cache = True,
do_sample=False,
repetition_penalty=1.2
)[0]
print(tokenizer.decode(output))