From 21634e53cc59b6094535c3ff82f23a44cb804775 Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Tue, 25 Feb 2025 14:15:28 +0100
Subject: [PATCH] zmiana modelu

---
 hft.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hft.py b/hft.py
index cc4f09d..f3f9e2c 100644
--- a/hft.py
+++ b/hft.py
@@ -151,7 +151,7 @@ class CustomTrainer(Trainer):
 
 # Inicjalizacja komponentów
 source_mapper = SourceMapper()
-model_name = "mistralai/Mistral-7B-v0.1" #"google/gemma-2-2b"
+model_name = "EleutherAI/gpt-neo-2.7B" #"google/gemma-2-2b"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 tokenizer.pad_token = tokenizer.eos_token