Instructions to use nvidia/AMPLIFY_120M with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nvidia/AMPLIFY_120M with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="nvidia/AMPLIFY_120M", trust_remote_code=True, device_map="auto")# Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("nvidia/AMPLIFY_120M", trust_remote_code=True, dtype="auto", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Add library name (#1)
Browse files- Add library name (4358a41bc2e76bec9f9e584884b4b2fbd770b091)
Co-authored-by: Niels Rogge <nielsr@users.noreply.huggingface.co>