psychedelicious
6ab84741a0
fix(nodes): make ModelsList an enum-keyed dict
...
The `ModelsList` OpenAPI schema is generated as being keyed by plain strings. This means that API consumers do not know the shape of the dict. It _should_ be keyed by the `SDModelType` enum.
Unfortunately, `fastapi` does not actually handle this correctly yet; it still generates the schema with plain string keys.
Adding this anyways though in hopes that it will be resolved upstream and we can get the correct schema. Until then, I'll implement the (simple but annoying) logic on the frontend.
https://github.com/pydantic/pydantic/issues/4393
2023-05-16 15:02:58 +10:00
Lincoln Stein
eea0d6f7bc
default to no filter in list_models()
2023-05-15 23:52:29 -04:00
Lincoln Stein
4fe94a9315
list_models() now returns a dict of {type,{name: info}}
2023-05-15 23:44:08 -04:00
Lincoln Stein
426f4eaf7e
adjusted regression tests to work with new SDModelTypes
2023-05-13 22:29:33 -04:00
Lincoln Stein
baf5451fa0
Merge branch 'main' into lstein/new-model-manager
2023-05-13 22:01:34 -04:00
Sergey Borisov
039fa73269
Change SDModelType enum to string, fixes(model unload negative locks count, scheduler load error, saftensors convert, wrong logic in del_model, wrong parse metadata in web)
2023-05-14 03:06:26 +03:00
Lincoln Stein
b23c9f1da5
get Tuple type hint syntax right
2023-05-13 14:59:21 -04:00
Lincoln Stein
5e8e3cf464
correct typos in model_manager_service
2023-05-13 14:55:59 -04:00
Lincoln Stein
72967bf118
convert add_model(), del_model(), list_models() etc to use bifurcated names
2023-05-13 14:44:44 -04:00
Sergey Borisov
bc96727cbe
Rewrite latent nodes to new model manager
2023-05-13 16:08:03 +03:00
Sergey Borisov
3b2a054f7a
Add model loader node; unet, clip, vae fields; change compel node to clip field
2023-05-13 04:37:20 +03:00
Eugene Brodsky
63db3fc22f
reduce queue check interval to 0.5s
2023-05-12 17:54:26 -04:00
Eugene
ad0bb3f61a
fix: queue error should not crash InvocationProcessor
...
1. if retrieving an item from the queue raises an exception, the
InvocationProcessor thread crashes, but the API continues running in
a non-functional state. This fixes the issue
2. when there are no items in the queue, sleep 1 second before checking
again.
3. Also ensures the thread isn't crashed if an exception is raised from
invoker, and emits the error event
Intentionally using base Exceptions because for now we don't know which
specific exception to expect.
Fixes (sort of)? #3222
2023-05-12 17:54:26 -04:00
Sergey Borisov
4492044d29
Redo compel node to separate model loading
2023-05-12 23:09:33 +03:00
Sergey Borisov
5431dd5f50
Fix event args
2023-05-12 23:08:03 +03:00
Sergey Borisov
79fecba274
Fix model manager initialization in web ui
2023-05-12 23:05:08 +03:00
psychedelicious
e5b7dd63e9
fix(nodes): temporarily disable librarygraphs
...
- Do not retrieve graph from DB until we resolve the issue of changing node schemas causing application to fail to start up due to invalid graphs
2023-05-12 22:33:49 +10:00
Lincoln Stein
2ef79b8bf3
fix bug in persistent model scheme
2023-05-12 00:14:56 -04:00
blessedcoolant
9c1e65f3a3
Merge branch 'main' into feat/nodes/add-randomintinvocation
2023-05-12 15:56:41 +12:00
psychedelicious
ebec200ba6
Remove unused import
2023-05-12 13:56:02 +10:00
Lincoln Stein
11ecf438f5
latents.py converted to use model manager service; events emitted
2023-05-11 23:33:24 -04:00
blessedcoolant
0acb8ed85d
Merge branch 'main' into feat/nodes/add-w-h-latentsoutput
2023-05-12 15:23:29 +12:00
blessedcoolant
8c1c9cd702
Merge branch 'main' into feat/nodes/add-randomintinvocation
2023-05-12 15:21:49 +12:00
blessedcoolant
af95cef7f9
Merge branch 'main' into fix/nodes/fix-imageoutput-optionals
2023-05-12 15:08:19 +12:00
Lincoln Stein
df5b968954
model manager now running as a service
2023-05-11 21:24:29 -04:00
Sergey Borisov
27dc07d95a
Set zero eta by default(fix ddim scheduler error)
2023-05-11 18:49:27 +03:00
blessedcoolant
f7dc171c4f
Rename default schedulers across the app
2023-05-12 03:44:20 +12:00
blessedcoolant
8a836247c8
Add DPMPP Single, Euler Karras and DPMPP2 Multi Karras Schedulers
2023-05-12 02:23:33 +12:00
blessedcoolant
9a383e456d
Codesplit SCHEDULER_MAP for reusage
2023-05-12 00:40:03 +12:00
blessedcoolant
c1e7460d39
Merge branch 'main' into unipc-sched
2023-05-12 00:11:09 +12:00
blessedcoolant
3ffff023b2
Add missing key to scheduler_map
...
It was breaking coz the sampler was not being reset. So needs a key on each. Will simplify this later.
2023-05-12 00:08:50 +12:00
psychedelicious
6cf308004a
fix(nodes): remove Optionals on ImageOutputs
2023-05-11 20:54:57 +10:00
blessedcoolant
d1029138d2
Default to DDIM if scheduler is missing
2023-05-11 22:54:35 +12:00
blessedcoolant
06b5800d28
Add UniPC Scheduler
2023-05-11 22:43:18 +12:00
psychedelicious
483f2ccb56
feat(nodes): add RandomIntInvocation
...
just outputs a single random int
2023-05-11 20:33:32 +10:00
psychedelicious
93ced0bec6
feat(nodes): add w/h to latents outputs
...
This reduces the number of nodes needed when working with latents (ie fewer plain integer value nodes)
Also correct a few mistakes in the fields
2023-05-11 20:32:55 +10:00
psychedelicious
4333852c37
fix(nodes): fix missing context arg in LatentsToLatents
2023-05-11 19:28:42 +10:00
Lincoln Stein
8ad8c5c67a
resolve conflicts with main
2023-05-11 00:19:20 -04:00
Lincoln Stein
590942edd7
Merge branch 'main' into lstein/new-model-manager
2023-05-11 00:16:03 -04:00
Lincoln Stein
4627910c5d
added a wrapper model_manager_service and model events
2023-05-11 00:09:19 -04:00
psychedelicious
f488b1a7f2
fix(nodes): fix usage of Optional
2023-05-11 11:55:51 +10:00
psychedelicious
34f3a0f0e3
feat(nodes): improve default model choosing output
2023-05-11 11:55:51 +10:00
psychedelicious
d0bac1675e
fix(nodes): fix ImageOutput Config
2023-05-11 11:55:51 +10:00
psychedelicious
4e56c962f4
fix(nodes): fix infill docstrings
2023-05-11 11:55:51 +10:00
psychedelicious
4ef0e43759
fix(nodes): remove dataURL invocation
2023-05-11 11:55:51 +10:00
psychedelicious
a7786d5ff2
fix(nodes): restore seamless to TextToLatents
2023-05-11 11:55:51 +10:00
psychedelicious
6c1de975d9
feat(nodes): add infill nodes
2023-05-11 11:55:51 +10:00
psychedelicious
a1079e455a
feat(nodes): cleanup unused params, seed generation
2023-05-11 11:55:51 +10:00
psychedelicious
da4eacdffe
feat(nodes): add InfillInvocation
2023-05-11 11:55:51 +10:00
psychedelicious
6102e560ba
feat(nodes): add LatentsToImage node (VAE encode)
2023-05-11 11:55:51 +10:00
