Set `library_name` to `tf-keras`.
Browse filesModel 'probing-vits/vit_b16_patch16_224_i21k_i1k' seems to be compatible only with "Keras 2" and not "Keras 3". To distinguish them, models compatible with legacy Keras 2.x should be tagged as `tf-keras` while models compatible with Keras 3.x are tagged as `keras`.
This PR updates the model card to replace the explicit `library_name: keras` metadata which is now outdated by `library_name: tf-keras`. Updating this metadata will facilitate its discoverability and usage.
For more information about `keras` and `tf-keras` library names, check out this pull request: https://github.com/huggingface/huggingface.js/pull/774.
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
library_name: keras
|
3 |
---
|
4 |
|
5 |
This model is a TensorFlow port of ViT B-16 [1] trained with recipes from [2]. It was first pre-trained on ImageNet-21k and was then fine-tuned on the ImageNet-1k dataset. You can refer to [this notebook](https://github.com/sayakpaul/probing-vits/blob/main/notebooks/load-jax-weights-vitb16.ipynb) to know how the porting was done.
|
|
|
1 |
---
|
2 |
+
library_name: tf-keras
|
3 |
---
|
4 |
|
5 |
This model is a TensorFlow port of ViT B-16 [1] trained with recipes from [2]. It was first pre-trained on ImageNet-21k and was then fine-tuned on the ImageNet-1k dataset. You can refer to [this notebook](https://github.com/sayakpaul/probing-vits/blob/main/notebooks/load-jax-weights-vitb16.ipynb) to know how the porting was done.
|