Use ErnieTEModel_ not ErnieTEModel.

This commit is contained in:
bedovyy 2026-04-16 20:06:51 +09:00
parent e9a2d1e4cc
commit 1648e352c6

View File

@ -35,4 +35,4 @@ def te(dtype_llama=None, llama_quantization_metadata=None):
model_options = model_options.copy() model_options = model_options.copy()
model_options["quantization_metadata"] = llama_quantization_metadata model_options["quantization_metadata"] = llama_quantization_metadata
super().__init__(device=device, dtype=dtype, model_options=model_options) super().__init__(device=device, dtype=dtype, model_options=model_options)
return ErnieTEModel return ErnieTEModel_