Prep for future breaking changes in newer versions of transformers library

2026-04-10 12:04:32 -06:00
parent 145144eee3
commit bfb373c8fa
4 changed files with 1 additions and 60 deletions
--- a/toolkit/reference_adapter.py
+++ b/toolkit/reference_adapter.py
@@ -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