pep8

2024-09-13 17:18:44 +02:00 · 2024-09-13 17:18:44 +02:00 · 82b35492af
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:
--- a/backend/open_webui/env.py
+++ b/backend/open_webui/env.py
@ -38,6 +38,7 @@ USE_CUDA = os.environ.get("USE_CUDA_DOCKER", "false")
 if USE_CUDA.lower() == "true":
    try:
        import torch
+
        assert torch.cuda.is_available(), "CUDA not available"
        DEVICE_TYPE = "cuda"
    except Exception as e: