Merge pull request #362 from langflow-ai/improve-gpu-detection

improve gpu detection

docker-compose-cpu.yml

@@ -63,8 +63,6 @@ services:
       - OPENSEARCH_USERNAME=admin
       - OPENSEARCH_PASSWORD=${OPENSEARCH_PASSWORD}
       - OPENAI_API_KEY=${OPENAI_API_KEY}
-      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
-      - NVIDIA_VISIBLE_DEVICES=all
       - GOOGLE_OAUTH_CLIENT_ID=${GOOGLE_OAUTH_CLIENT_ID}
       - GOOGLE_OAUTH_CLIENT_SECRET=${GOOGLE_OAUTH_CLIENT_SECRET}
       - MICROSOFT_GRAPH_OAUTH_CLIENT_ID=${MICROSOFT_GRAPH_OAUTH_CLIENT_ID}

src/tui/screens/monitor.py

@@ -2,7 +2,7 @@
 
 import asyncio
 import re
-from typing import Literal, Any
+from typing import Literal, Any, Optional
 
 # Define button variant type
 ButtonVariant = Literal["default", "primary", "success", "warning", "error"]
@@ -296,7 +296,7 @@ class MonitorScreen(Screen):
                 self.run_worker(self._show_logs(service_name))
                 self._start_follow(service_name)
 
-    async def _start_services(self, cpu_mode: bool = False) -> None:
+    async def _start_services(self, cpu_mode: Optional[bool] = None) -> None:
         """Start services with progress updates."""
         self.operation_in_progress = True
         try: