Skip to content

fix: add Hugging Face API key to StableDiffusionV3 parameters and remove deprecated model option#283

Merged
cristian-tamblay merged 3 commits into
developfrom
fix/image-generation
Sep 5, 2025
Merged

fix: add Hugging Face API key to StableDiffusionV3 parameters and remove deprecated model option#283
cristian-tamblay merged 3 commits into
developfrom
fix/image-generation

Conversation

@Irozuku
Copy link
Copy Markdown
Collaborator

@Irozuku Irozuku commented Sep 3, 2025

This pull request updates the Stable Diffusion v3 model integration in DashAI to support loading private models from Hugging Face by allowing users to provide an API key. It also improves error handling during authentication and model loading. Additionally, it removes an unused model option from the schema.

Authentication and API key support:

  • Added a new huggingface_key field to the StableDiffusionSchema to allow users to specify their Hugging Face API key for accessing private models.
  • Updated the model initialization to authenticate with Hugging Face using the provided API key, with error handling for failed logins.
  • Imported the login function from huggingface_hub to enable authentication.

Model selection schema update:

  • Removed "stabilityai/stable-diffusion-3-medium-tensorrt" from the list of available model options in the schema, likely because it is deprecated or unsupported.

Robust model loading:

  • Improved error handling when loading the selected model, raising a clear error if the model cannot be loaded.

@cristian-tamblay cristian-tamblay merged commit 35fe955 into develop Sep 5, 2025
5 checks passed
@cristian-tamblay cristian-tamblay deleted the fix/image-generation branch September 5, 2025 17:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants