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

-3
View File
@@ -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