File size: 782 Bytes
a15fec5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | """
Trendyol DinoV2 Image Similarity Model
This package contains a fine-tuned DinoV2 model for e-commerce image similarity.
Fully compatible with Hugging Face transformers.
"""
from .modeling_trendyol_dinov2 import TrendyolDinoV2Model, TrendyolDinoV2Config
from .image_processing_trendyol_dinov2 import TrendyolDinoV2ImageProcessor
# Register for AutoModel and AutoImageProcessor
from transformers import AutoConfig, AutoModel, AutoImageProcessor
AutoConfig.register("trendyol_dinov2", TrendyolDinoV2Config)
AutoModel.register(TrendyolDinoV2Config, TrendyolDinoV2Model)
AutoImageProcessor.register(TrendyolDinoV2Config, TrendyolDinoV2ImageProcessor)
__version__ = "1.0.0"
__all__ = [
"TrendyolDinoV2Model",
"TrendyolDinoV2Config",
"TrendyolDinoV2ImageProcessor"
]
|