UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 05:24:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

pep8

Browse Source
This commit is contained in:
thiswillbeyourgithub 2024-09-13 17:18:44 +02:00
parent a7ea07036e
commit 82b35492af
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/open_webui/__init__.py
View File

@ -41,6 +41,7 @@ def serve(
) )
try: try:
import torch import torch
assert torch.cuda.is_available(), "CUDA not available" assert torch.cuda.is_available(), "CUDA not available"
typer.echo("CUDA seems to be working") typer.echo("CUDA seems to be working")
except Exception as e: except Exception as e:

1
backend/open_webui/env.py
View File

@ -38,6 +38,7 @@ USE_CUDA = os.environ.get("USE_CUDA_DOCKER", "false")
if USE_CUDA.lower() == "true": if USE_CUDA.lower() == "true":
try: try:
import torch import torch
assert torch.cuda.is_available(), "CUDA not available" assert torch.cuda.is_available(), "CUDA not available"
DEVICE_TYPE = "cuda" DEVICE_TYPE = "cuda"
except Exception as e: except Exception as e: