visheratin
/

nllb-siglip-mrl-base

Zero-Shot Image Classification

Model card Files Files and versions Community

visheratin commited on Mar 5

Commit

7b7b6ab

•

1 Parent(s): 50bbe59

Update nllb_mrl.py

Files changed (1) hide show

nllb_mrl.py +2 -3

nllb_mrl.py CHANGED Viewed

@@ -54,9 +54,8 @@ class MatryoshkaNllbClip(PreTrainedModel):
         self.model = create_model(
             config.clip_model_name, output_dict=True
         )
-        pretrained_config = get_pretrained_cfg(config.clip_model_name, "v1")
-        print(pretrained_config)
-        pp_cfg = PreprocessCfg(pretrained_config["preprocess_cfg"])
         self.transform = image_transform_v2(
             pp_cfg,
             is_train=False,

         self.model = create_model(
             config.clip_model_name, output_dict=True
         )
+        preprocess_cfg = get_pretrained_cfg(config.clip_model_name, "v1")
+        pp_cfg = PreprocessCfg(preprocess_cfg)
         self.transform = image_transform_v2(
             pp_cfg,
             is_train=False,