POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mod

This commit is contained in:
l.gabrysiak 2025-02-25 14:46:27 +01:00
parent 30234c332a
commit fa04f1faa0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hft.py
View File

@ -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 # Zmiana importu
from transformers import GPTNeoForCausalLM, Trainer, TrainingArguments, AutoTokenizer # Zmiana importu
from datasets import Dataset
from PIL import Image
import re