pep8

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:

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: