Skip to content

Commit 1648e35

Browse files
committed
Use ErnieTEModel_ not ErnieTEModel.
1 parent e9a2d1e commit 1648e35

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/text_encoders/ernie.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,4 +35,4 @@ def __init__(self, device="cpu", dtype=None, model_options={}):
3535
model_options = model_options.copy()
3636
model_options["quantization_metadata"] = llama_quantization_metadata
3737
super().__init__(device=device, dtype=dtype, model_options=model_options)
38-
return ErnieTEModel
38+
return ErnieTEModel_

0 commit comments

Comments
 (0)