Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add install probes for T5_encoder and ClipTextModel #6800

Merged
merged 4 commits into from
Sep 3, 2024

Conversation

lstein
Copy link
Collaborator

@lstein lstein commented Aug 31, 2024

Summary

During testing of the new Flux starter models, I found that installation of clip-vit-large and the quantized text encoders were failing during the model installer's model probe phase. This PR adds model probe support for the FLUX T5_encoder and ClipTextModel dependencies. This PR fixes "download failed" errors experienced when the FLUX starter models are installed.

Related Issues / Discussions

QA Instructions

To confirm the original issue, please first remove the T5 quantized text encoder and clip-vit-large, and try to install one of the quantized Flux models from the Model manager starter models tab. This should fail.

After pulling this PR, these tasks should run to completion and flux-based workflows should work.

Merge Plan

Merge when approved.

Checklist

  • The PR has a short but descriptive title, suitable for a changelog
  • Tests added / updated (if applicable)
  • Documentation added / updated (if applicable)

@lstein lstein changed the title Add probes for T5_encoder and ClipTextModel Add install probes for T5_encoder and ClipTextModel Aug 31, 2024
@github-actions github-actions bot added python PRs that change python files backend PRs that change backend files labels Aug 31, 2024
@lstein lstein requested a review from brandonrising September 3, 2024 14:10
@brandonrising brandonrising force-pushed the lstein/feat/probe-flux-models branch from facca8c to 4a20ce5 Compare September 3, 2024 19:47
@brandonrising brandonrising enabled auto-merge (rebase) September 3, 2024 19:47
@brandonrising brandonrising merged commit 6dabe4d into main Sep 3, 2024
14 checks passed
@brandonrising brandonrising deleted the lstein/feat/probe-flux-models branch September 3, 2024 19:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
backend PRs that change backend files python PRs that change python files
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants