From 229841e05e4e77f39986d87e62b3688455dc773b Mon Sep 17 00:00:00 2001 From: Jamie Pine Date: Thu, 29 Jan 2026 03:12:11 -0800 Subject: [PATCH] Add GPU type information to health check response - Updated the health check endpoint to include the type of GPU available (CUDA or MPS). - Modified the HealthResponse model to accommodate the new gpu_type field, enhancing the response with detailed GPU information. - This change improves the clarity of system capabilities for users and developers. --- backend/main.py | 7 +++++++ backend/models.py | 1 + 2 files changed, 8 insertions(+) diff --git a/backend/main.py b/backend/main.py index 7a9b1500..e3f6909b 100644 --- a/backend/main.py +++ b/backend/main.py @@ -79,6 +79,12 @@ async def health(): has_mps = hasattr(torch.backends, 'mps') and torch.backends.mps.is_available() gpu_available = has_cuda or has_mps + gpu_type = None + if has_cuda: + gpu_type = f"CUDA ({torch.cuda.get_device_name(0)})" + elif has_mps: + gpu_type = "MPS (Apple Silicon)" + vram_used = None if has_cuda: vram_used = torch.cuda.memory_allocated() / 1024 / 1024 # MB @@ -136,6 +142,7 @@ async def health(): model_downloaded=model_downloaded, model_size=model_size, gpu_available=gpu_available, + gpu_type=gpu_type, vram_used_mb=vram_used, ) diff --git a/backend/models.py b/backend/models.py index 624f7f6a..b0f4452d 100644 --- a/backend/models.py +++ b/backend/models.py @@ -118,6 +118,7 @@ class HealthResponse(BaseModel): model_downloaded: Optional[bool] = None # Whether model is cached/downloaded model_size: Optional[str] = None # Current model size if loaded gpu_available: bool + gpu_type: Optional[str] = None # GPU type (CUDA, MPS, or None) vram_used_mb: Optional[float] = None