mod
This commit is contained in:
parent
41b757409f
commit
b519db2b27
2
hft.py
2
hft.py
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue