-
Notifications
You must be signed in to change notification settings - Fork 2.7k
[MM2] Use typed ModelRecordChanges for model_install() rather than untyped dict #6645
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
Conversation
…ModelRecordChanges - adjusted frontend to work with new schema - used this facility to assign "starter model" names and descriptions to the installed models.
…ModelRecordChanges - adjusted frontend to work with new schema - used this facility to assign "starter model" names and descriptions to the installed models.
8a3617e
to
3ed47de
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good! I was just wishing for this today when I installed the SwinIR
upscaling model, which has a fifty-thousand character name by default.
I'll request a review from @brandonrising too.
.../src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StartModelsResultItem.tsx
Outdated
Show resolved
Hide resolved
.../src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StartModelsResultItem.tsx
Outdated
Show resolved
Hide resolved
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
the backend changes lgtm!
…ai/InvokeAI into lstein/feat/starter-descriptions
…ModelRecordChanges - adjusted frontend to work with new schema - used this facility to assign "starter model" names and descriptions to the installed models.
…ModelRecordChanges - adjusted frontend to work with new schema - used this facility to assign "starter model" names and descriptions to the installed models.
#6645 inadvertently removed the lockfile
#6645 inadvertently removed the lockfile
Summary
The various model manager install calls previously accepted a dict object called
config
containing fields to override the autoprobed model configuration values. This PR replaces this with a typed pydantic objectModelRecordChanges
. This is the same class that is used for model record updates.This PR takes advantage of this change to write the name and description of the curated "starter models" into the model record database. Note that this is my first foray into typescript, and may not have implemented the necessary changes in the most elegant way.
Related Issues / Discussions
Closes #6528
QA Instructions
After running the back and frontends, try the following:
Merge Plan
Merge when approved.
Checklist