Instructions to use nvidia/MambaVision-L2-512-21K with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nvidia/MambaVision-L2-512-21K with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="nvidia/MambaVision-L2-512-21K", trust_remote_code=True) pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoModelForImageClassification model = AutoModelForImageClassification.from_pretrained("nvidia/MambaVision-L2-512-21K", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Improve model card: Set pipeline tag, add library name, fix license link
#1
by nielsr HF Staff - opened
README.md
CHANGED
|
@@ -1,13 +1,13 @@
|
|
| 1 |
---
|
| 2 |
-
license: other
|
| 3 |
-
license_name: nvclv1
|
| 4 |
-
license_link: LICENSE
|
| 5 |
datasets:
|
| 6 |
- ILSVRC/imagenet-21k
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
-
|
| 11 |
[**MambaVision: A Hybrid Mamba-Transformer Vision Backbone**](https://arxiv.org/abs/2407.08083).
|
| 12 |
|
| 13 |
## Model Overview
|
|
|
|
| 1 |
---
|
|
|
|
|
|
|
|
|
|
| 2 |
datasets:
|
| 3 |
- ILSVRC/imagenet-21k
|
| 4 |
+
license: other
|
| 5 |
+
license_name: nvclv1
|
| 6 |
+
license_link: https://huggingface.co/nvidia/MambaVision-L2-512-21K/blob/main/LICENSE
|
| 7 |
+
pipeline_tag: image-classification
|
| 8 |
+
library_name: transformers
|
| 9 |
---
|
| 10 |
|
|
|
|
| 11 |
[**MambaVision: A Hybrid Mamba-Transformer Vision Backbone**](https://arxiv.org/abs/2407.08083).
|
| 12 |
|
| 13 |
## Model Overview
|