mod
This commit is contained in:
parent
41b757409f
commit
b519db2b27
2
hft.py
2
hft.py
|
|
@ -1,7 +1,7 @@
|
||||||
import os
|
import os
|
||||||
import torch
|
import torch
|
||||||
import torch.nn as nn
|
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 datasets import Dataset
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
import re
|
import re
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue