ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove vision model from qwen text encoder as it is not needed for image generation currently

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-06 11:40:02 -06:00
parent 14ccf2f3ce
commit 4c4a10d439
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions_built_in/diffusion_models/qwen_image/qwen_image.py
View File

@@ -130,6 +130,10 @@ class QwenImageModel(BaseModel):
text_encoder = Qwen2_5_VLForConditionalGeneration.from_pretrained( text_encoder = Qwen2_5_VLForConditionalGeneration.from_pretrained(
base_model_path, subfolder="text_encoder", torch_dtype=dtype base_model_path, subfolder="text_encoder", torch_dtype=dtype
) )
# remove the visual model as it is not needed for image generation
text_encoder.model.visual = None
text_encoder.to(self.device_torch, dtype=dtype) text_encoder.to(self.device_torch, dtype=dtype)
flush() flush()