pep8

2025-12-16 03:47:49 +01:00 · 2024-09-13 17:18:44 +02:00
parent a7ea07036e
commit 82b35492af
2 changed files with 2 additions and 0 deletions
--- a/backend/open_webui/init.py
+++ b/backend/open_webui/init.py
@@ -41,6 +41,7 @@ def serve(
        )
        try:
            import torch
+
            assert torch.cuda.is_available(), "CUDA not available"
            typer.echo("CUDA seems to be working")
        except Exception as e: