move text encoder 2 loading
This commit is contained in:
parent
6b6f63564e
commit
d11b37f0b2
|
@ -369,16 +369,7 @@ def load_text_encoders(
|
|||
)
|
||||
text_encoder_session._model_path = path.join(model, "text_encoder")
|
||||
components["text_encoder_session"] = text_encoder_session
|
||||
else:
|
||||
components["text_encoder"] = OnnxRuntimeModel(
|
||||
OnnxRuntimeModel.load_model(
|
||||
text_encoder.SerializeToString(),
|
||||
provider=device.ort_provider("text-encoder"),
|
||||
sess_options=text_encoder_opts,
|
||||
)
|
||||
)
|
||||
|
||||
if params.is_xl():
|
||||
text_encoder_2 = path.join(model, "text_encoder_2", ONNX_MODEL)
|
||||
text_encoder_2 = blend_loras(
|
||||
server,
|
||||
|
@ -404,6 +395,14 @@ def load_text_encoders(
|
|||
)
|
||||
text_encoder_2_session._model_path = path.join(model, "text_encoder_2")
|
||||
components["text_encoder_2_session"] = text_encoder_2_session
|
||||
else:
|
||||
components["text_encoder"] = OnnxRuntimeModel(
|
||||
OnnxRuntimeModel.load_model(
|
||||
text_encoder.SerializeToString(),
|
||||
provider=device.ort_provider("text-encoder"),
|
||||
sess_options=text_encoder_opts,
|
||||
)
|
||||
)
|
||||
|
||||
return components
|
||||
|
||||
|
|
Loading…
Reference in New Issue