diff --git a/hft.py b/hft.py index cc97948..f83d558 100644 --- a/hft.py +++ b/hft.py @@ -151,7 +151,7 @@ class CustomTrainer(Trainer): # Inicjalizacja komponentów source_mapper = SourceMapper() -model_name = "Sao10K/Fimbulvetr-11B-v2-GGUF" #"google/gemma-2-2b" +model_name = "mistralai/Mistral-7B-v0.1" #"google/gemma-2-2b" tokenizer = AutoTokenizer.from_pretrained(model_name) tokenizer.pad_token = tokenizer.eos_token