Use `ErnieTEModel_` not `ErnieTEModel`. (#13431)

2026-04-16 23:11:58 +09:00 · 2026-04-16 23:11:58 +09:00 · b41ab53b6f
parent e9a2d1e4cc
commit b41ab53b6f
1 changed files with 1 additions and 1 deletions
--- a/comfy/text_encoders/ernie.py
+++ b/comfy/text_encoders/ernie.py
@ -35,4 +35,4 @@ def te(dtype_llama=None, llama_quantization_metadata=None):
                model_options = model_options.copy()
                model_options["quantization_metadata"] = llama_quantization_metadata
            super().__init__(device=device, dtype=dtype, model_options=model_options)
-    return ErnieTEModel
+    return ErnieTEModel_