Instructions to use nvidia/MambaVision-S-1K with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nvidia/MambaVision-S-1K with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-feature-extraction", model="nvidia/MambaVision-S-1K", trust_remote_code=True)# Load model directly from transformers import AutoModelForImageClassification model = AutoModelForImageClassification.from_pretrained("nvidia/MambaVision-S-1K", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Update pipeline tag, add library name, and link to code
#1
by nielsr HF Staff - opened
This PR updates the pipeline_tag to image-classification to better reflect the model's primary use case. It also adds the library_name as transformers since the Hugging Face Transformers library is used in the provided examples. It also ensures that it is easier for users to find the official code repository: https://github.com/NVlabs/MambaVision.