rydlrKE commited on
Commit
9a526ac
·
verified ·
1 Parent(s): 5430059

fix: prefer local text encoder API in api mode when available

Browse files
Files changed (1) hide show
  1. kimodo/model/load_model.py +9 -0
kimodo/model/load_model.py CHANGED
@@ -200,6 +200,15 @@ def _select_text_encoder_conf(text_encoder_url: str) -> dict:
200
  if mode == "local":
201
  return _build_local_text_encoder_conf()
202
  if mode == "api":
 
 
 
 
 
 
 
 
 
203
  try:
204
  _probe_api_text_encoder(text_encoder_url, autostart_enabled)
205
  return _build_api_text_encoder_conf(text_encoder_url)
 
200
  if mode == "local":
201
  return _build_local_text_encoder_conf()
202
  if mode == "api":
203
+ if (
204
+ not _is_local_text_encoder_url(text_encoder_url)
205
+ and local_api_url
206
+ and _is_local_text_encoder_url(local_api_url)
207
+ and _is_port_open(local_api_url)
208
+ ):
209
+ print(f"Using local text encoder API at {local_api_url} (remote URL also configured).")
210
+ _probe_api_text_encoder(local_api_url, autostart_enabled=False)
211
+ return _build_api_text_encoder_conf(local_api_url)
212
  try:
213
  _probe_api_text_encoder(text_encoder_url, autostart_enabled)
214
  return _build_api_text_encoder_conf(text_encoder_url)