Spaces:
Running on Zero
Running on Zero
fix: prefer local text encoder API in api mode when available
Browse files
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)
|