mix nasty.models.download (Nasty v0.3.0)
View SourceDownloads a pre-trained transformer model from HuggingFace.
Usage
mix nasty.models.download MODEL_NAME [OPTIONS]Arguments
- MODEL_NAME - Name of the model to download (e.g., roberta_base, bert_base_cased)
Options
--cache-dir- Directory to cache models (default: priv/models/transformers)--offline- Use only cached models, don't download (default: false)
Examples
# Download RoBERTa base model
mix nasty.models.download roberta_base
# Download BERT to custom directory
mix nasty.models.download bert_base_cased --cache-dir=/tmp/models
# Download XLM-RoBERTa for multilingual support
mix nasty.models.download xlm_roberta_baseAvailable Models
- bert_base_cased - BERT base (110M params, English)
- bert_base_uncased - BERT base uncased (110M params, English)
- roberta_base - RoBERTa base (125M params, English, recommended)
- xlm_roberta_base - XLM-RoBERTa (270M params, 100 languages)
- distilbert_base - DistilBERT (66M params, English, fast)