Instructions to use nvidia/MambaVision-T2-1K with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nvidia/MambaVision-T2-1K with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-feature-extraction", model="nvidia/MambaVision-T2-1K", trust_remote_code=True)# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForImageClassification model = AutoModelForImageClassification.from_pretrained("nvidia/MambaVision-T2-1K", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Update model card: set pipeline tag, add library name and citation
Verified signatureThis PR fixes the pipeline tag, setting it to `image-classification` as that is the main usecase of this model. It also adds the `library_name` tag, so that the "how to use" button appears on the top right of the model page. Finally, citation information and star history are added for better discoverability.71f01aa
-
Update README.md
Verified signature07f24ef
-
Update README.md
Verified signature8ca44d0 -
Update README.md
Verified signaturec0c45ae -
Update config.json
Verified signaturef4a7aa1 -
Update modeling_mambavision.py
Verified signature76ac99c -
Upload model
Verified signature1ab9b66 -
Upload model
Verified signature93ce109
-
Create mamba_vision.py
Verified signaturee3785c4 -
Update README.md
Verified signatureede60d3 -
Create hf_model.py
Verified signature29ac3ef -
Create config.json
Verified signaturefe1315a
-
Update LICENSE
Verified signature03f8a47 -
Upload mambavision_tiny2_1k.pth.tar
Verified signatureb6a44ee -
initial commit
Verified signature4d20db7