visheratin commited on
Commit
50bbe59
·
verified ·
1 Parent(s): 58a7420

Update nllb_mrl.py

Browse files
Files changed (1) hide show
  1. 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,