Update README.md
Browse files
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 |
-
|
| 63 |
-
|
|
|
|
| 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()
|