Instructions to use nvidia/MambaVision-L2-1K with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nvidia/MambaVision-L2-1K with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="nvidia/MambaVision-L2-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-L2-1K", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Update pipeline tag, add Github link and library name
Verified signatureThis PR: - Updates the pipeline tag from `image-feature-extraction` to `image-classification` for better searchability. - Adds the Github link to the model card. - Adds the `library_name` to let people know it's compatible with the `transformers` library.3cf3e60
-
Update README.md
Verified signaturea04c551
-
Update README.md
Verified signaturee029823 -
Update README.md
Verified signature2d3b1af -
Update config.json
Verified signature91e7203 -
Upload model
Verified signature584e818 -
Upload model
Verified signature7e65190 -
Delete hf_model.py
Verified signature8a2c177
-
Update README.md
Verified signature2f4b387 -
Create hf_model.py
Verified signaturee8ac58c -
Create mamba_vision.py
Verified signaturec63e1cd -
Create config.json
Verified signaturedca3fec
-
Update LICENSE
Verified signature08e03e6 -
Upload mambavision_large2_1k.pth.tar
Verified signature3c2984d -
initial commit
Verified signature63005a7