diff --git a/hft.py b/hft.py
index 9827899..391ff4a 100644
--- a/hft.py
+++ b/hft.py
@@ -100,7 +100,7 @@ class CustomTrainer(Trainer):
         return (loss, outputs) if return_outputs else loss
 
 # Przygotowanie modelu i tokenizera
-model_name = "google/gemma-2b"
+model_name = "google/gemma-2-2b"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = CustomModel.from_pretrained(model_name)