Update nllb_mrl.py
Browse files- nllb_mrl.py +1 -0
nllb_mrl.py
CHANGED
@@ -55,6 +55,7 @@ class MatryoshkaNllbClip(PreTrainedModel):
|
|
55 |
config.clip_model_name, output_dict=True
|
56 |
)
|
57 |
pretrained_config = get_pretrained_cfg(config.clip_model_name, "v1")
|
|
|
58 |
pp_cfg = PreprocessCfg(pretrained_config["preprocess_cfg"])
|
59 |
self.transform = image_transform_v2(
|
60 |
pp_cfg,
|
|
|
55 |
config.clip_model_name, output_dict=True
|
56 |
)
|
57 |
pretrained_config = get_pretrained_cfg(config.clip_model_name, "v1")
|
58 |
+
print(pretrained_config)
|
59 |
pp_cfg = PreprocessCfg(pretrained_config["preprocess_cfg"])
|
60 |
self.transform = image_transform_v2(
|
61 |
pp_cfg,
|