google
/

vit-base-patch32-224-in21k

Image Feature Extraction

Model card Files Files and versions Community

nielsr HF staff commited on Dec 8, 2022

Commit

ee39d2d

·

1 Parent(s): a657cce

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ outputs = model(**inputs)
 last_hidden_state = outputs.last_hidden_state
 ```
-Currently, both the feature extractor and model  support PyTorch. Tensorflow and JAX/FLAX are coming soon, and the API of ViTFeatureExtractor might change.
 ## Training data

 last_hidden_state = outputs.last_hidden_state
 ```
+Refer to the [docs](https://huggingface.co/docs/transformers/model_doc/vit) for usage in TensorFlow and JAX/FLAX.
 ## Training data