zmiana modelu
This commit is contained in:
parent
0121c1bf68
commit
84f8adfdbd
2
hft.py
2
hft.py
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue