diff --git a/hft.py b/hft.py index fa94e42..cc97948 100644 --- a/hft.py +++ b/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