Instructions to use UFNLP/gatortron-medium with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use UFNLP/gatortron-medium with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("UFNLP/gatortron-medium", dtype="auto") - Notebooks
- Google Colab
- Kaggle
ValueError: MegatronBertForSequenceClassification does not support `device_map='auto'` yet.
#1
by shanchen - opened
is int8 support available?
These are questions for the transformer team I think. If they have not implemented to support these features for Megatron, then you need to raise the issues to the transformer repository. @shanchen
it seems other people did this. it didn't work w/ MegatronBertForSequenceClassification
https://huggingface.co/mosaicml/mpt-7b/discussions/45/files
Hello, how can I use GartoTron for QA task ?