Prep for future breaking changes in newer versions of transformers library
This commit is contained in:
1 parent
145144eee3
commit
bfb373c8fa
4 files changed
+1
-60
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user