diff --git a/bridge/server.py b/bridge/server.py index daf2a8a..285c868 100644 --- a/bridge/server.py +++ b/bridge/server.py @@ -150,12 +150,17 @@ def _ensure_brain(): compute = os.environ.get("WHISPER_COMPUTE_TYPE", "auto") try: whisper = WhisperModel(cfg.whisper_model, device=device, compute_type=compute) + # Log the device actually resolved by CTranslate2 (device="auto" + # picks cuda when available) so a silent CPU load is visible. + resolved = str(getattr(getattr(whisper, "model", None), "device", device)).lower() + print(f"[bridge] whisper loaded on {resolved} (compute={compute})", flush=True) except Exception as ge: # GPU not available / unsupported -> fall back to CPU so the # bridge still works without a GPU passed to the container. if device != "cpu": print(f"[bridge] whisper device='{device}' failed ({ge}); falling back to CPU", flush=True) whisper = WhisperModel(cfg.whisper_model, device="cpu", compute_type="int8") + print("[bridge] whisper loaded on cpu (compute=int8)", flush=True) else: raise