Commit ·
66a47bb
1
Parent(s): 301e2d7
Update configuration_vtde.py
Browse files- configuration_vtde.py +1 -0
configuration_vtde.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
from transformers import VisionTextDualEncoderConfig
|
| 2 |
|
| 3 |
class VTDEConfig(VisionTextDualEncoderConfig):
|
|
|
|
| 4 |
def __init__(self, projection_dim=512, logit_scale_init_value=2.6592,
|
| 5 |
text_pooling_mode='mean',
|
| 6 |
vision_pooling_mode='max',
|
|
|
|
| 1 |
from transformers import VisionTextDualEncoderConfig
|
| 2 |
|
| 3 |
class VTDEConfig(VisionTextDualEncoderConfig):
|
| 4 |
+
model_type = "vtde"
|
| 5 |
def __init__(self, projection_dim=512, logit_scale_init_value=2.6592,
|
| 6 |
text_pooling_mode='mean',
|
| 7 |
vision_pooling_mode='max',
|