mix ragex.models.download (Ragex v0.8.0)

View Source

Downloads and caches Bumblebee embedding models for offline use.

This task downloads all configured embedding models from HuggingFace and stores them in the Bumblebee cache directory. This is useful for:

  • Building Docker images with pre-cached models
  • Offline/air-gapped environments
  • Faster startup times (no download delay)

Usage

# Download default model only
mix ragex.models.download

# Download all available models
mix ragex.models.download --all

# Download specific model(s)
mix ragex.models.download --models all_minilm_l6_v2,codebert_base

# Use custom cache directory
mix ragex.models.download --cache-dir /path/to/cache

Options

  • --all - Download all available models from the registry
  • --models - Comma-separated list of model IDs to download
  • --cache-dir - Custom cache directory (overrides BUMBLEBEE_CACHE_DIR)
  • --quiet - Suppress informational output

Cache Location

Models are cached in:

  • Custom: Directory specified by --cache-dir or BUMBLEBEE_CACHE_DIR env var
  • Default: Platform-specific cache directory (~/.cache/bumblebee on Linux)

Use Bumblebee.cache_dir() to see the active cache location.