Only apply gemma quant config to gemma model for newbie. (#11436)

2025-12-19 22:02:43 -08:00
parent 4c432c11ed
commit fb478f679a
2 changed files with 6 additions and 1 deletions
--- a/comfy/text_encoders/newbie.py
+++ b/comfy/text_encoders/newbie.py
@@ -57,6 +57,6 @@ def te(dtype_llama=None, llama_quantization_metadata=None):
        def __init__(self, device="cpu", dtype=None, model_options={}):
            if llama_quantization_metadata is not None:
                model_options = model_options.copy()
-                model_options["quantization_metadata"] = llama_quantization_metadata
+                model_options["llama_quantization_metadata"] = llama_quantization_metadata
            super().__init__(dtype_gemma=dtype_llama, device=device, dtype=dtype, model_options=model_options)
    return NewBieTEModel_