diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-27 09:02:22 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-27 09:02:22 +0300 |
| commit | 68f336bd994bed5442ad95bad6b6ad5564a5409a (patch) | |
| tree | c49a1dabb84c89d2c5dbac61ea67042ea01d0e76 /modules/sd_hijack.py | |
| parent | a3ddf464a2ed24c999f67ddfef7969f8291567be (diff) | |
| parent | 50973ec77c297edc3b3c581e871b970dde1af8ba (diff) | |
Merge branch 'release_candidate'
Diffstat (limited to 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index f5615967..c8fdd4f1 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -243,7 +243,7 @@ class StableDiffusionModelHijack: ldm.modules.diffusionmodules.openaimodel.UNetModel.forward = sd_unet.UNetModel_forward
def undo_hijack(self, m):
- if type(m.cond_stage_model) == xlmr.BertSeriesModelWithTransformation:
+ if type(m.cond_stage_model) == sd_hijack_xlmr.FrozenXLMREmbedderWithCustomWords:
m.cond_stage_model = m.cond_stage_model.wrapped
elif type(m.cond_stage_model) == sd_hijack_clip.FrozenCLIPEmbedderWithCustomWords:
|