zmiana modelu

This commit is contained in:
l.gabrysiak 2025-02-25 14:03:55 +01:00
parent 0121c1bf68
commit 84f8adfdbd
1 changed files with 1 additions and 1 deletions

2
hft.py
View File

@ -151,7 +151,7 @@ class CustomTrainer(Trainer):
# Inicjalizacja komponentów
source_mapper = SourceMapper()
model_name = "perplexity-ai/r1-1776"#"google/gemma-2-2b"
model_name = "Sao10K/Fimbulvetr-11B-v2-GGUF" #"google/gemma-2-2b"
tokenizer = AutoTokenizer.from_pretrained(model_name)
tokenizer.pad_token = tokenizer.eos_token