Instructions to use nvidia/MambaVision-T-1K with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nvidia/MambaVision-T-1K with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="nvidia/MambaVision-T-1K", trust_remote_code=True) pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForImageClassification model = AutoModelForImageClassification.from_pretrained("nvidia/MambaVision-T-1K", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +0 -4
config.json
CHANGED
|
@@ -2,10 +2,6 @@
|
|
| 2 |
"architectures": [
|
| 3 |
"MambaVisionModelForImageClassification"
|
| 4 |
],
|
| 5 |
-
"auto_map": {
|
| 6 |
-
"AutoConfig": "",
|
| 7 |
-
"AutoModelForImageClassification": ""
|
| 8 |
-
},
|
| 9 |
"depths": [
|
| 10 |
1,
|
| 11 |
3,
|
|
|
|
| 2 |
"architectures": [
|
| 3 |
"MambaVisionModelForImageClassification"
|
| 4 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
"depths": [
|
| 6 |
1,
|
| 7 |
3,
|