ML4RS-Anonymous commited on
Commit
17f0e8d
·
verified ·
1 Parent(s): 66ee1b2

fix: import create_model_from_pretrained

Browse files
Files changed (1) hide show
  1. models/siglip_model.py +1 -1
models/siglip_model.py CHANGED
@@ -39,7 +39,7 @@ class SigLIPModel:
39
  if not os.path.exists(self.ckpt_path):
40
  print(f"Warning: Checkpoint not found at {self.ckpt_path}")
41
 
42
- if 'hf' in self.ckpt_path:
43
  from open_clip import create_model_from_pretrained, get_tokenizer # works on open-clip-torch>=2.23.0, timm>=0.9.8
44
 
45
  model, preprocess = create_model_from_pretrained('hf-hub:timm/ViT-SO400M-14-SigLIP-384')
 
39
  if not os.path.exists(self.ckpt_path):
40
  print(f"Warning: Checkpoint not found at {self.ckpt_path}")
41
 
42
+ # if 'hf' in self.ckpt_path:
43
  from open_clip import create_model_from_pretrained, get_tokenizer # works on open-clip-torch>=2.23.0, timm>=0.9.8
44
 
45
  model, preprocess = create_model_from_pretrained('hf-hub:timm/ViT-SO400M-14-SigLIP-384')