Commit Graph

3134 Commits

Author SHA1 Message Date
psychedelicious
393e32f8a7 chore(ui): typegen 2024-02-26 14:49:38 -05:00
Mary Hipp
30fdb9dbfd undo 2024-02-26 14:44:37 -05:00
Mary Hipp
66f6013436 fix literal strings in MM UI 2024-02-26 14:44:37 -05:00
Mary Hipp
49b04f7db8 fix TI appearing as key in prompt 2024-02-26 14:20:28 -05:00
Mary Hipp
253dc5d43d fix base model grouping in combobox 2024-02-26 14:20:28 -05:00
psychedelicious
dfa5505ed8 feat(ui): bulk download click to download 2024-02-25 22:23:15 -05:00
psychedelicious
f8b731b900 fix(ui): fix node types for canvas graphs 2024-02-24 19:38:16 +11:00
psychedelicious
fd9ab0fb7d chore(ui): typegen 2024-02-24 19:38:16 +11:00
blessedcoolant
afe6639b9c fix: outpaint result not getting pasted back correctly 2024-02-24 19:38:16 +11:00
blessedcoolant
8fa238f100 canvas: improve paste back (or try to) 2024-02-24 19:38:16 +11:00
blessedcoolant
30b6a0ee23 wip(ui): Replace 2 Layer Coherence pass with Gradient Mask 2024-02-24 19:38:16 +11:00
psychedelicious
b51b163400 fix(ui): fix merge issue 2024-02-24 19:04:52 +11:00
psychedelicious
7e13224ec8 fix(ui): use new scan_folder response instead of hook to determine if models are installed already 2024-02-24 19:04:52 +11:00
psychedelicious
7bc454209c chore(ui): typegen 2024-02-24 19:04:52 +11:00
psychedelicious
8b8d950137 chore(ui): lint 2024-02-24 19:04:52 +11:00
psychedelicious
24fd7f41ff build(ui): restore i18n eslint rule 2024-02-24 19:04:52 +11:00
psychedelicious
a5dba4b0d9 fix(ui): fix metadata route 2024-02-24 19:04:52 +11:00
psychedelicious
72fb1cefff chore(ui): typegen 2024-02-24 19:04:52 +11:00
Mary Hipp
974658107d lint fix 2024-02-24 19:04:52 +11:00
Mary Hipp
07fb5d5c19 updated translations 2024-02-24 19:04:52 +11:00
Mary Hipp
cfcb68696c clean up old model manager components and endpoints 2024-02-24 19:04:52 +11:00
Mary Hipp
7b1b6d3235 add model convert to checkpoint main models 2024-02-24 19:04:52 +11:00
Mary Hipp
aefba52a0a fix logic to see if scanned models are already installed, style tweaks 2024-02-24 19:04:52 +11:00
Mary Hipp
190702d011 add error_reason to UI if import fails 2024-02-24 19:04:52 +11:00
Mary Hipp
7785e8ff79 fix types for ImportQueue, add QuickAdd for scan models 2024-02-24 19:04:52 +11:00
Jennifer Player
b3beaefa04 refactored and fixed issues with advanced import form 2024-02-24 19:04:52 +11:00
psychedelicious
98be81354a fix(ui): misc MM cleanup 2024-02-24 19:04:52 +11:00
psychedelicious
2a2a5eb775 chore(ui): temp disable eslint i18 rule 2024-02-24 19:04:52 +11:00
psychedelicious
4a42b15b42 fix(ui): fix ImportMainModelResponse type 2024-02-24 19:04:52 +11:00
psychedelicious
f24d5e5e31 fix(ui): simplify model install event listeners 2024-02-24 19:04:52 +11:00
psychedelicious
4b106bc903 fix(ui): fix model install event types 2024-02-24 19:04:52 +11:00
Jennifer Player
135ef9066f added advanced import forms, not fully working yet 2024-02-24 19:04:52 +11:00
Mary Hipp
0567f98e4a get positioning/scrolling working for scan results list 2024-02-24 19:04:52 +11:00
Mary Hipp
5b66baa3ec basic scan working and renders results 2024-02-24 19:04:52 +11:00
Mary Hipp
a022aaf258 add scan model endpoint, break add model into tabs 2024-02-24 19:04:52 +11:00
Mary Hipp
94065b090a update metadata endpoint 2024-02-24 19:04:52 +11:00
Jennifer Player
8d243b1fca added status to import queue model 2024-02-24 19:04:52 +11:00
Jennifer Player
23c412e011 delete model imports and prune all finished, update state with socket messages 2024-02-24 19:04:52 +11:00
Mary Hipp
66692f02aa fix sync model endpoint 2024-02-24 19:04:52 +11:00
Mary Hipp
38af1c3a81 form error handling 2024-02-24 19:04:52 +11:00
Mary Hipp
7b4b7e3781 finish model update 2024-02-24 19:04:52 +11:00
Jennifer Player
02a3472505 added socket listeners, added more info to ui 2024-02-24 19:04:52 +11:00
Mary Hipp
909d354a38 edit view for model, depending on type and valid values 2024-02-24 19:04:52 +11:00
Mary Hipp
7801b8c42f hook up Add Model button 2024-02-24 19:04:52 +11:00
Mary Hipp
4fd259bb89 single model view 2024-02-24 19:04:52 +11:00
Jennifer Player
b8b3ef9725 added import model form and importqueue 2024-02-24 19:04:52 +11:00
Mary Hipp
3a8d5dc349 model list, filtering, searching 2024-02-24 19:04:52 +11:00
Mary Hipp
358cac9674 workspace for mary and jenn 2024-02-24 19:04:52 +11:00
Mary Hipp
bdc2b8069b get old UI working somewhat with new endpoints 2024-02-24 19:04:52 +11:00
psychedelicious
d103ff0d6e fix(ui): roll back utility-types
It's `Required` util does not distribute over unions as expected. Also we have `ts-toolbelt` already for some utils.
2024-02-23 07:53:45 +11:00
psychedelicious
94931e8ac0 feat(ui): refactor metadata handling
Refactor of metadata recall handling. This is in preparation for a backwards compatibility layer for models.

- Create helpers to fetch a model outside react (e.g. not in a hook)
- Created helpers to parse model metadata
- Renamed a lot of types that were confusing and/or had naming collisions
2024-02-23 07:53:45 +11:00
psychedelicious
b409f3aaf9 chore(ui): typegen 2024-02-23 07:53:45 +11:00
psychedelicious
de3be4bd30 feat(ui): replace type-fest with utility-types
- The new package has more useful types
- Only used `JsonObject` from `type-fest`; added an implementation of that type
2024-02-23 07:53:45 +11:00
Lincoln Stein
ae14df97d6 remove startup dependency on legacy models.yaml file 2024-02-23 07:47:39 +11:00
Brandon Rising
c0d9990344 Create /search endpoint, update model object structure in scan model page 2024-02-22 09:50:02 -05:00
psychedelicious
4ac5e307c4 chore(ui): bump deps
Notable updates:
- Minor version of RTK includes customizable selectors for RTK Query, so we can remove the patch that was added to ensure only the LRU memoize function was used for perf reasons. Updated to use the LRU memoize function.
- Major version of react-resizable-panels. No breaking changes, works great, and you can now resize all panels when dragging at the intersection point of panels. Cool!
- Minor (?) version of nanostores. `action` API is removed, we were using it in one spot. Fixed.
- @invoke-ai/eslint-config-react has all deps bumped and now has its dependent plugins/configs listed as normal dependencies (as opposed to peer deps). This means we can remove those packages from explicit dev deps.
2024-02-22 07:27:28 +11:00
psychedelicious
2815f737fe tidy(ui): remove debugging stmt 2024-02-22 07:26:47 +11:00
psychedelicious
63e96fd1ea fix(ui): handle new model format for metadata 2024-02-22 07:26:47 +11:00
psychedelicious
66ab56246a fix(ui): use model names in badges 2024-02-22 07:26:47 +11:00
psychedelicious
82925e1539 fix(ui): fix package build 2024-02-21 08:31:55 -05:00
psychedelicious
0137a0db7b feat(ui): do not subscribe to bulk download sio room if baseUrl is set 2024-02-21 00:00:25 +11:00
psychedelicious
b410793684 feat(ui): revise bulk download listeners
- Use a single listener for all of the to keep them in one spot
- Use the bulk download item name as a toast id so we can update the existing toasts
- Update handling to work with other environments
- Move all bulk download handling from components to listener
2024-02-21 00:00:25 +11:00
psychedelicious
894e9f127b chore(ui): typegen 2024-02-21 00:00:25 +11:00
Stefan Tobler
8d9c566656 implementing download for bulk_download events 2024-02-21 00:00:25 +11:00
Stefan Tobler
9db7e073a3 setting up event listeners for bulk download socket 2024-02-21 00:00:25 +11:00
psychedelicious
0d0a2a5c91 fix(ui): get workflow editor model selects working 2024-02-20 13:33:31 +11:00
psychedelicious
0cab636ab0 fix(ui): get refiner model select working 2024-02-20 13:33:31 +11:00
psychedelicious
de097ec58a fix(ui): get vae model select working 2024-02-20 13:33:31 +11:00
psychedelicious
bb6f426162 fix(ui): get embedding select working 2024-02-20 13:33:31 +11:00
psychedelicious
663f135b3c fix(ui): get lora select working 2024-02-20 13:33:31 +11:00
psychedelicious
2f2097662a chore(ui): bump @invoke-ai/ui-library 2024-02-20 13:33:31 +11:00
psychedelicious
458c29cfa5 fix(ui): fix low-hanging fruit types 2024-02-20 13:33:31 +11:00
psychedelicious
fe04f28841 chore(ui): typegen 2024-02-20 09:48:14 +11:00
Lincoln Stein
b06d63fb34 remove errant def that was crashing invokeai-configure 2024-02-19 17:31:53 +11:00
Lincoln Stein
2ad0752582 Tidy names and locations of modules
- Rename old "model_management" directory to "model_management_OLD" in order to catch
  dangling references to original model manager.
- Caught and fixed most dangling references (still checking)
- Rename lora, textual_inversion and model_patcher modules
- Introduce a RawModel base class to simplfy the Union returned by the
  model loaders.
- Tidy up the model manager 2-related tests. Add useful fixtures, and
  a finalizer to the queue and installer fixtures that will stop the
  services and release threads.
2024-02-19 08:16:56 +11:00
Lincoln Stein
ba1f8878dd Fix issues identified during PR review by RyanjDick and brandonrising
- ModelMetadataStoreService is now injected into ModelRecordStoreService
  (these two services are really joined at the hip, and should someday be merged)
- ModelRecordStoreService is now injected into ModelManagerService
- Reduced timeout value for the various installer and download wait*() methods
- Introduced a Mock modelmanager for testing
- Removed bare print() statement with _logger in the install helper backend.
- Removed unused code from model loader init file
- Made `locker` a private variable in the `LoadedModel` object.
- Fixed up model merge frontend (will be deprecated anyway!)
2024-02-19 08:16:56 +11:00
psychedelicious
c7904a32f4 chore(ui): lint 2024-02-16 22:42:15 +11:00
psychedelicious
17f5484f5b feat(ui): fix main model & control adapter model selects 2024-02-16 22:41:09 +11:00
psychedelicious
86a372b02f refactor(ui): url builders for each router
The MM2 router is at `api/v2/models`. URL builder utils make this a bit easier to manage.
2024-02-16 21:57:30 +11:00
psychedelicious
2e9aa9391d feat(ui): update model identifier to be key (wip)
- Update most model identifiers to be `{key: string}` instead of name/base/type. Doesn't change the model select components yet.
- Update model _parameters_, stored in redux, to be `{key: string, base: BaseModel}` - we need to store the base model to be able to check model compatibility. May want to store the whole config? Not sure...
2024-02-16 18:56:02 +11:00
psychedelicious
0c8112cf28 fix(ui): update model types 2024-02-15 22:17:16 +11:00
psychedelicious
019898c7be tests(ui): add type tests 2024-02-15 22:16:55 +11:00
psychedelicious
2b1ff8d196 tests(ui): enable vitest type testing
This is useful for the zod schemas and types we have created to match the backend.
2024-02-15 22:16:11 +11:00
psychedelicious
79fb691b4d chore(ui): typegen 2024-02-15 22:15:21 +11:00
psychedelicious
560ae17e21 feat(ui): export components type 2024-02-15 21:16:25 +11:00
psychedelicious
2bd1ab2f1c fix(ui): fix type issues 2024-02-15 20:53:41 +11:00
psychedelicious
ed43472582 chore: lint 2024-02-15 20:52:44 +11:00
Lincoln Stein
94e8d1b6d5 make model manager v2 ready for PR review
- Replace legacy model manager service with the v2 manager.

- Update invocations to use new load interface.

- Fixed many but not all type checking errors in the invocations. Most
  were unrelated to model manager

- Updated routes. All the new routes live under the route tag
  `model_manager_v2`. To avoid confusion with the old routes,
  they have the URL prefix `/api/v2/models`. The old routes
  have been de-registered.

- Added a pytest for the loader.

- Updated documentation in contributing/MODEL_MANAGER.md
2024-02-15 18:00:08 +11:00
Lincoln Stein
d959276217 fix invokeai_configure script to work with new mm; rename CLIs 2024-02-15 17:56:01 +11:00
Lincoln Stein
dfcf38be91 BREAKING CHANGES: invocations now require model key, not base/type/name
- Implement new model loader and modify invocations and embeddings

- Finish implementation loaders for all models currently supported by
  InvokeAI.

- Move lora, textual_inversion, and model patching support into
  backend/embeddings.

- Restore support for model cache statistics collection (a little ugly,
  needs work).

- Fixed up invocations that load and patch models.

- Move seamless and silencewarnings utils into better location
2024-02-15 17:56:01 +11:00
Lincoln Stein
fbded1c0f2 Multiple refinements on loaders:
- Cache stat collection enabled.
- Implemented ONNX loading.
- Add ability to specify the repo version variant in installer CLI.
- If caller asks for a repo version that doesn't exist, will fall back
  to empty version rather than raising an error.
2024-02-15 17:51:07 +11:00
psychedelicious
f505ec64ba tests(ui): add parseFieldType.test.ts 2024-02-15 17:32:38 +11:00
psychedelicious
f22eb368a3 feat(ui): add more types of FieldParseError
Unfortunately you cannot test for both a specific type of error and match its message. Splitting the error classes makes it easier to test expected error conditions.
2024-02-15 17:32:38 +11:00
psychedelicious
96ae22c7e0 feat(ui): add vitest
- Add vitest.
- Consolidate vite configs into single file (easier to config everything based on env for testing)
2024-02-15 17:32:38 +11:00
psychedelicious
f5447cdc23 feat(ui): workflow schema v3 (WIP)
The changes aim to deduplicate data between workflows and node templates, decoupling workflows from internal implementation details. A good amount of data that was needlessly duplicated from the node template to the workflow is removed.

These changes substantially reduce the file size of workflows (and therefore the images with embedded workflows):

- Default T2I SD1.5 workflow JSON is reduced from 23.7kb (798 lines) to 10.9kb (407 lines).
- Default tiled upscale workflow JSON is reduced from 102.7kb (3341 lines) to 51.9kb (1774 lines).

The trade-off is that we need to reference node templates to get things like the field type and other things. In practice, this is a non-issue, because we need a node template to do anything with a node anyways.

- Field types are not included in the workflow. They are always pulled from the node templates.

The field type is now properly an internal implementation detail and we can change it as needed. Previously this would require a migration for the workflow itself. With the v3 schema, the structure of a field type is an internal implementation detail that we are free to change as we see fit.

- Workflow nodes no long have an `outputs` property and there is no longer such a thing as a `FieldOutputInstance`. These are only on the templates.

These were never referenced at a time when we didn't also have the templates available, and there'd be no reason to do so.

- Node width and height are no longer stored in the node.

These weren't used. Also, per https://reactflow.dev/api-reference/types/node, we shouldn't be programmatically changing these properties. A future enhancement can properly add node resizing.

- `nodeTemplates` slice is merged back into `nodesSlice` as `nodes.templates`. Turns out it's just a hassle having these separate in separate slices.

- Workflow migration logic updated to support the new schema. V1 workflows migrate all the way to v3 now.

- Changes throughout the nodes code to accommodate the above changes.
2024-02-15 17:32:38 +11:00
psychedelicious
c76a6bd65f chore(ui): regen types 2024-02-15 17:30:03 +11:00
psychedelicious
8e2b61e19f feat(ui): revise graphs to not use LinearUIOutputInvocation
See this comment for context: https://github.com/invoke-ai/InvokeAI/pull/5491#discussion_r1480760629

- Remove this now-unnecessary node from all graphs
- Update graphs' terminal image-outputting nodes' `is_intermediate` and `board` fields appropriately
- Add util function to prepare the `board` field, tidy the utils
- Update `socketInvocationComplete` listener to work correctly with this change

I've manually tested all graph permutations that were changed (I think this is all...) to ensure images go to the gallery as expected:
- ad-hoc upscaling
- t2i w/ sd1.5
- t2i w/ sd1.5 & hrf
- t2i w/ sdxl
- t2i w/ sdxl + refiner
- i2i w/ sd1.5
- i2i w/ sdxl
- i2i w/ sdxl + refiner
- canvas t2i w/ sd1.5
- canvas t2i w/ sdxl
- canvas t2i w/ sdxl + refiner
- canvas i2i w/ sd1.5
- canvas i2i w/ sdxl
- canvas i2i w/ sdxl + refiner
- canvas inpaint w/ sd1.5
- canvas inpaint w/ sdxl
- canvas inpaint w/ sdxl + refiner
- canvas outpaint w/ sd1.5
- canvas outpaint w/ sdxl
- canvas outpaint w/ sdxl + refiner
2024-02-15 17:30:02 +11:00
psychedelicious
a3faa3792a chore(ui): regen types 2024-02-15 17:30:02 +11:00
psychedelicious
e36d925bce fix(ui): remove original l2i node in HRF graph 2024-02-15 17:30:02 +11:00
chainchompa
d3acb81743
Merge branch 'main' into chainchompa/reorder-exposed-fields 2024-02-14 18:26:35 -05:00