github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Prep for future breaking changes in newer versions of transformers library

Browse Source
This commit is contained in:
Jaret Burkett
2026-04-10 12:04:32 -06:00
parent 145144eee3
commit bfb373c8fa
4 changed files with 1 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
jobs/process/BaseSDTrainProcess.py
View File

@@ -19,8 +19,7 @@ from safetensors.torch import save_file, load_file
from torch.utils.data import DataLoader
import torch
import torch.backends.cuda
from huggingface_hub import HfApi, Repository, interpreter_login
from huggingface_hub.utils import HfFolder
from huggingface_hub import HfApi, interpreter_login
from toolkit.memory_management import MemoryManager
from toolkit.basic import value_map