Prep for future breaking changes in newer versions of transformers library

This commit is contained in:
Jaret Burkett committed 2026-04-10 12:04:32 -06:00
1 parent 145144eee3
commit bfb373c8fa
4 files changed
+1 -60

No files matched your search

+1 -2
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