Refactor qwen5b model code to be qwen 5b specific

2025-08-06 10:54:56 -06:00
parent 5d8922fca2
commit 14ccf2f3ce
4 changed files with 5 additions and 5 deletions
--- a/extensions_built_in/diffusion_models/qwen_image/qwen_image.py
+++ b/extensions_built_in/diffusion_models/qwen_image/qwen_image.py
@@ -136,7 +136,7 @@ class QwenImageModel(BaseModel):
        if self.model_config.quantize_te:
            self.print_and_status_update("Quantizing Text Encoder")
            quantize(text_encoder, weights=get_qtype(
-                self.model_config.qtype))
+                self.model_config.qtype_te))
            freeze(text_encoder)
            flush()