ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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: