Skip to content

Add support for new Hugging Face models (Llama, Mistral, SDXL variants) and related dependencies#500

Merged
cristian-tamblay merged 16 commits into
developfrom
feat/add-generative-models
Mar 26, 2026
Merged

Add support for new Hugging Face models (Llama, Mistral, SDXL variants) and related dependencies#500
cristian-tamblay merged 16 commits into
developfrom
feat/add-generative-models

Conversation

@Irozuku
Copy link
Copy Markdown
Collaborator

@Irozuku Irozuku commented Mar 25, 2026

Summary

Introduces support for multiple new Hugging Face models (e.g., Llama, Mistral, Mixtral, SDXL variants) by implementing their classes and registering them in the initial component registry, making them available in DashAI. As part of these integrations, additional dependencies were introduced, and noisy third-party startup warnings from these libraries are now suppressed to maintain a clean console output.


Type of Change

Check all that apply like this [x]:

  • Backend change
  • Frontend change
  • CI / Workflow change
  • Build / Packaging change
  • Bug fix
  • Documentation

Changes (by file)

  • DashAI/back/models/...: Added multiple new model classes and schemas (Llama, Mistral, Mixtral, PixArtSigma, SDXL variants, ControlNet variants, etc.).
  • DashAI/back/initial_components.py: Updated get_initial_components to register all newly supported models.
  • DashAI/__main__.py: Suppressed noisy warnings introduced by new model dependencies.
  • DashAI/back/dependencies/job_queues/huey_job_queue.py: Suppressed warnings in background jobs caused by newly added libraries.

Testing (optional)

  • Verify that all newly added models appear in the model selection UI and can be instantiated correctly.
  • Run a sample inference for at least one model per family (e.g., Llama, Mistral, SDXL) to confirm proper integration.
  • Confirm that startup logs remain clean despite the addition of new dependencies.

Notes (optional)

  • Warning suppression targets noise introduced by dependencies required for the new models and does not affect core application logging.

Irozuku added 14 commits March 24, 2026 16:23
Replace en dashes, multiplication signs, and inconsistent
range formatting with hyphens and 'x', standardizing all
numeric ranges and resolution descriptions in English and
Spanish docstrings.
… with respective schemas for image generation
…-Image models with respective schemas for image generation
…olLM, PixArtSigma, SD15 Depth, HED, OpenPose, SDXL Canny, SDXL Turbo, and Stable Diffusion XL with respective imports
…dling

Move heavyweight/optional imports into methods/constructors to reduce startup cost.
Raise RuntimeError only when missing dependencies are actually used, with clearer messages and tracebacks.
…ai weights

Replace stabilityai/stable-diffusion-2* model IDs with sd2-community/* equivalents; update description in EN and ES to note that the original weights have been deprecated and removed from the stabilityai organization.
@Irozuku Irozuku marked this pull request as ready for review March 25, 2026 17:32
@Irozuku Irozuku added back Backend work enhancement New feature or request labels Mar 25, 2026
@cristian-tamblay cristian-tamblay merged commit ea1d803 into develop Mar 26, 2026
19 checks passed
@cristian-tamblay cristian-tamblay deleted the feat/add-generative-models branch March 26, 2026 17:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

back Backend work enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants