add demofusion

This commit is contained in:
Vladimir Mandic
2023-12-09 14:15:48 -05:00
parent f91735deac
commit 1b1ae6784c
6 changed files with 1298 additions and 7 deletions
+1 -1
View File
@@ -60,7 +60,7 @@ class DiffusersTextualInversionManager(BaseTextualInversionManager):
def encode_prompts(pipe, prompts: list, negative_prompts: list, clip_skip: typing.Optional[int] = None):
if 'StableDiffusion' not in pipe.__class__.__name__:
if 'StableDiffusion' not in pipe.__class__.__name__ and 'DemoFusion':
shared.log.warning(f"Prompt parser not supported: {pipe.__class__.__name__}")
return None, None, None, None
else: