From 5c0e4381e748ec74b1d0d11d1a03369a016eddf8 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Tue, 25 Feb 2025 14:06:10 +0100 Subject: [PATCH] zmiana modelu --- hft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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