diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-11 18:16:43 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-11 18:16:43 +0000 |
| commit | af081211ee93622473ee575de30fed2fd8263c09 (patch) | |
| tree | b047d051d31e8332f6a8491c41bbcb4c35ccf692 /modules/sd_hijack.py | |
| parent | 7b833291b3ef4690ef158ee3415c2e93948acb2d (diff) | |
| download | stable-diffusion-webui-gfx803-af081211ee93622473ee575de30fed2fd8263c09.tar.gz stable-diffusion-webui-gfx803-af081211ee93622473ee575de30fed2fd8263c09.tar.bz2 stable-diffusion-webui-gfx803-af081211ee93622473ee575de30fed2fd8263c09.zip | |
getting SD2.1 to run on SDXL repo
Diffstat (limited to 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index 3b6f95ce..c4b9211f 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -166,6 +166,15 @@ class StableDiffusionModelHijack: undo_optimizations()
def hijack(self, m):
+ conditioner = getattr(m, 'conditioner', None)
+ if conditioner:
+ for i in range(len(conditioner.embedders)):
+ embedder = conditioner.embedders[i]
+ if type(embedder).__name__ == 'FrozenOpenCLIPEmbedder':
+ embedder.model.token_embedding = EmbeddingsWithFixes(embedder.model.token_embedding, self)
+ m.cond_stage_model = sd_hijack_open_clip.FrozenOpenCLIPEmbedderWithCustomWords(embedder, self)
+ conditioner.embedders[i] = m.cond_stage_model
+
if type(m.cond_stage_model) == xlmr.BertSeriesModelWithTransformation:
model_embeddings = m.cond_stage_model.roberta.embeddings
model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.word_embeddings, self)
|