diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index a904aa36..f51b191c 100644 --- a/api/onnx_web/diffusers/load.py +++ b/api/onnx_web/diffusers/load.py @@ -383,7 +383,7 @@ def load_text_encoders( # session for te1 text_encoder_session = InferenceSession( - text_encoder, + text_encoder.SerializeToString(), providers=[device.ort_provider("text-encoder")], sess_options=text_encoder_opts, ) @@ -392,7 +392,7 @@ def load_text_encoders( # session for te2 text_encoder_2_session = InferenceSession( - text_encoder_2, + text_encoder_2.SerializeToString(), providers=[device.ort_provider("text-encoder")], sess_options=text_encoder_2_opts, ) @@ -402,7 +402,7 @@ def load_text_encoders( # session for te components["text_encoder"] = OnnxRuntimeModel( OnnxRuntimeModel.load_model( - text_encoder, + text_encoder.SerializeToString(), provider=device.ort_provider("text-encoder"), sess_options=text_encoder_opts, )