Skip to content

Commit 317b5eb

Browse files
authored
Add support for LCM main models (invoke-ai#5152)
## What type of PR is this? (check all applicable) - [ ] Refactor - [X] Feature - [ ] Bug Fix - [ ] Optimization - [ ] Documentation Update - [ ] Community Node Submission ## Have you discussed this change with the InvokeAI team? - [ ] Yes - [X] No, because: Small obvious fix ## Have you updated all relevant documentation? - [X] Yes - [ ] No ## Description This one-line patch adds support for LCM models such as `SimianLuo/LCM_Dreamshaper_v7` ## Related Tickets & Documents <!-- For pull requests that relate or close an issue, please include them below. For example having the text: "closes invoke-ai#1234" would connect the current pull request to issue 1234. And when we merge the pull request, Github will automatically close the issue. --> - Closes invoke-ai#4951 ## QA Instructions, Screenshots, Recordings Try installing `SimianLuo/LCM_Dreamshaper_v7` and using with CFG 2.5 and the LCM scheduler. <!-- Please provide steps on how to test changes, any hardware or software specifications as well as any other pertinent information. --> ## Added/updated tests? - [ ] Yes - [X] Not needed
2 parents 1a596a5 + 98a4930 commit 317b5eb

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

invokeai/backend/model_management/model_probe.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ class ModelProbe(object):
5353
"StableDiffusionXLPipeline": ModelType.Main,
5454
"StableDiffusionXLImg2ImgPipeline": ModelType.Main,
5555
"StableDiffusionXLInpaintPipeline": ModelType.Main,
56+
"LatentConsistencyModelPipeline": ModelType.Main,
5657
"AutoencoderKL": ModelType.Vae,
5758
"AutoencoderTiny": ModelType.Vae,
5859
"ControlNetModel": ModelType.ControlNet,

0 commit comments

Comments
 (0)