From 745446f6fd77569373303d8c7848ad70fbec8948 Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Tue, 25 Feb 2025 14:46:55 +0100
Subject: [PATCH] mod

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

diff --git a/hft.py b/hft.py
index f423943..756535e 100644
--- a/hft.py
+++ b/hft.py
@@ -2,7 +2,7 @@ import os
 import torch
 import torch.nn as nn
 #from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments, Trainer
-from transformers import GPTNeoForCausalLM, Trainer, TrainingArguments, AutoTokenizer # Zmiana importu
+from transformers import GPTNeoForCausalLM, Trainer, TrainingArguments, AutoTokenizer, AutoModelForCausalLM # Zmiana importu
 from datasets import Dataset
 from PIL import Image
 import re