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"
]