This commit is contained in:
l.gabrysiak 2025-02-25 14:45:00 +01:00
parent 41b757409f
commit b519db2b27
1 changed files with 1 additions and 1 deletions

2
hft.py
View File

@ -1,7 +1,7 @@
import os
import torch
import torch.nn as nn
from transformers import AutoTokenizer, GPTNeoForCausalLM # Poprawiono importy
from transformers import AutoTokenizer, GPTNeoForCausalLM, Trainer # Poprawiono importy
from datasets import Dataset
from PIL import Image
import re