Update nllb_mrl.py

nllb_mrl.py

@@ -4,6 +4,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from open_clip import create_model, get_tokenizer
+from open_clip.pretrained import get_pretrained_cfg
 from open_clip.transform import PreprocessCfg, image_transform_v2
 from PIL import Image
 from transformers import PretrainedConfig, PreTrainedModel
@@ -53,7 +54,8 @@ class MatryoshkaNllbClip(PreTrainedModel):
         self.model = create_model(
             config.clip_model_name, output_dict=True
         )
-        pp_cfg = PreprocessCfg(**self.model.visual.preprocess_cfg)
+        pretrained_config = get_pretrained_cfg(config.clip_model_name, "v1")
+        pp_cfg = PreprocessCfg(pretrained_config.preprocess_cfg)
         self.transform = image_transform_v2(
             pp_cfg,
             is_train=False,