diff --git a/comfy/text_encoders/ernie.py b/comfy/text_encoders/ernie.py
index 2c7df78fe537..46d24d2229c9 100644
--- a/comfy/text_encoders/ernie.py
+++ b/comfy/text_encoders/ernie.py
@@ -35,4 +35,4 @@ def __init__(self, device="cpu", dtype=None, model_options={}):
                 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_