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
@@ -30,9 +30,6 @@ from transformers import (
|
||||
)
|
||||
from toolkit.models.size_agnostic_feature_encoder import SAFEImageProcessor, SAFEVisionModel
|
||||
|
||||
from transformers import ViTHybridImageProcessor, ViTHybridForImageClassification
|
||||
|
||||
from transformers import ViTFeatureExtractor, ViTForImageClassification
|
||||
|
||||
import torch.nn.functional as F
|
||||
import torch.nn as nn
|
||||
|
||||
Reference in new issue
Block a user