fixes

onyx-dot-app · Feb 8, 2025 · fb931ee · fb931ee
1 parent bc2c56d
commit fb931ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backend/model_server/encoders.py b/backend/model_server/encoders.py
@@ -412,7 +412,7 @@ async def embed_text(
             f"event=embedding_model "
             f"texts={len(texts)} "
             f"chars={total_chars} "
-            f"model={provider_type} "
+            f"model={model_name} "
             f"gpu={gpu_type} "
             f"elapsed={elapsed:.2f}"
         )

diff --git a/backend/model_server/main.py b/backend/model_server/main.py
@@ -60,7 +60,7 @@ def _move_files_recursively(source: Path, dest: Path, overwrite: bool = False) -
 @asynccontextmanager
 async def lifespan(app: FastAPI) -> AsyncGenerator:
     gpu_type = get_gpu_type()
-    logger.notice(f"gpu_type={gpu_type}")
+    logger.notice(f"Torch GPU Detection: gpu_type={gpu_type}")
 
     app.state.gpu_type = gpu_type