mlx-vision
/

vit_base_patch16_224.swag_lin-mlxim

Image Classification

Model card Files Files and versions Community

riccardomusmeci commited on Mar 20

Commit

aca2166

•

1 Parent(s): ca9e154

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -58,7 +58,7 @@ embeds = model(x)
 model = create_model("vit_base_patch16_224.swag_lin")
 model.eval()
-embeds = model.features(x)
 ```

 model = create_model("vit_base_patch16_224.swag_lin")
 model.eval()
+embeds = model.get_features(x)
 ```