CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

aio

Browse Source
This commit is contained in:
Haoming 2025-11-21 15:27:43 +08:00
parent 983a0ea79b
commit 1c637ad4f9
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/diffusion_engine/qwen.py
View File

@ -23,7 +23,7 @@ class QwenImage(ForgeDiffusionEngine):
super().__init__(estimated_config, huggingface_components)
self.is_inpaint = False
clip = CLIP(model_dict={"qwen25": huggingface_components["text_encoder"]}, tokenizer_dict={"qwen25": huggingface_components["tokenizer"]})
clip = CLIP(model_dict={"qwen25_7b": huggingface_components["text_encoder"]}, tokenizer_dict={"qwen25_7b": huggingface_components["tokenizer"]})
vae = VAE(model=huggingface_components["vae"], is_wan=True)
vae.first_stage_model.latent_format = self.model_config.latent_format
@ -33,8 +33,8 @@ class QwenImage(ForgeDiffusionEngine):
unet = UnetPatcher.from_model(model=huggingface_components["transformer"], diffusers_scheduler=None, k_predictor=k_predictor, config=estimated_config)
self.text_processing_engine_qwen = QwenTextProcessingEngine(
text_encoder=clip.cond_stage_model.qwen25,
tokenizer=clip.tokenizer.qwen25,
text_encoder=clip.cond_stage_model.qwen25_7b,
tokenizer=clip.tokenizer.qwen25_7b,
)
self.forge_objects = ForgeObjects(unet=unet, clip=clip, vae=vae, clipvision=None)

2
backend/loader.py
View File

@ -533,7 +533,7 @@ def replace_state_dict(sd: dict[str, torch.Tensor], asd: dict[str, torch.Tensor]
weight = asd["model.layers.0.self_attn.k_proj.bias"]
assert weight.shape[0] == 512
for k, v in asd.items():
sd[f"{text_encoder_key_prefix}qwen25.{k}"] = v
sd[f"{text_encoder_key_prefix}qwen25_7b.{k}"] = v
if "model.layers.0.post_feedforward_layernorm.weight" in asd:
assert "model.layers.0.self_attn.q_norm.weight" not in asd

2
modules_forge/packages/huggingface_guess/model_list.py
View File

@ -428,7 +428,7 @@ class QwenImage(BASE):
return ModelType.FLOW
def clip_target(self, state_dict: dict):
return {"qwen25": "text_encoder"}
return {"qwen25_7b": "text_encoder"}
models = [