This commit is contained in:
Richard Kuo (Danswer) 2025-02-07 17:28:17 -08:00
parent bc2c56dfb6
commit fb931ee4de
2 changed files with 2 additions and 2 deletions

View File

@ -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}"
)

View File

@ -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