mobileclip
fartashf commited on
Commit
4512a12
·
verified ·
1 Parent(s): 1ff441a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -59,8 +59,9 @@ import open_clip
59
  from PIL import Image
60
  from mobileclip.modules.common.mobileone import reparameterize_model
61
 
62
- model, _, preprocess = open_clip.create_model_and_transforms('MobileCLIP2-S3', pretrained='/path/to/mobileclip_s3.pt')
63
- tokenizer = open_clip.get_tokenizer('MobileCLIP2-S3')
 
64
 
65
  # Model needs to be in eval mode for inference because of batchnorm layers unlike ViTs
66
  model.eval()
 
59
  from PIL import Image
60
  from mobileclip.modules.common.mobileone import reparameterize_model
61
 
62
+ model_kwargs = {"image_mean": (0, 0, 0), "image_std": (1, 1, 1)}
63
+ model, _, preprocess = open_clip.create_model_and_transforms('MobileCLIP-S3', pretrained='/path/to/mobileclip_s3.pt', **model_kwargs)
64
+ tokenizer = open_clip.get_tokenizer('MobileCLIP-S3')
65
 
66
  # Model needs to be in eval mode for inference because of batchnorm layers unlike ViTs
67
  model.eval()