support altdiffusion-m18 (f8f4ff2b) · Commits · github_fork / Stable Diffusion Webui

modules/sd_hijack.py

+2 −0

Original line number	Diff line number	Diff line
		@@ -212,6 +212,7 @@ class StableDiffusionModelHijack:
		model_embeddings = m.cond_stage_model.roberta.embeddings
		model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.word_embeddings, self)
		m.cond_stage_model = sd_hijack_xlmr.FrozenXLMREmbedderWithCustomWords(m.cond_stage_model, self)

		elif type(m.cond_stage_model) == ldm.modules.encoders.modules.FrozenCLIPEmbedder:
		model_embeddings = m.cond_stage_model.transformer.text_model.embeddings
		model_embeddings.token_embedding = EmbeddingsWithFixes(model_embeddings.token_embedding, self)
		@@ -257,6 +258,7 @@ class StableDiffusionModelHijack:

		if hasattr(m, 'cond_stage_model'):
		delattr(m, 'cond_stage_model')

		elif type(m.cond_stage_model) == sd_hijack_xlmr.FrozenXLMREmbedderWithCustomWords:
		m.cond_stage_model = m.cond_stage_model.wrapped