gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Update main.py

Browse source
This commit is contained in:
Eric Hare 2025-09-04 09:41:56 -07:00
parent 15707a3228
commit da55c9e607
No known key found for this signature in database
GPG key ID: A73DF73724270AB7
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/main.py
View file

@ -56,8 +56,8 @@ from api import (
# Set multiprocessing start method to 'spawn' for CUDA compatibility # Set multiprocessing start method to 'spawn' for CUDA compatibility
multiprocessing.set_start_method("spawn", force=True) multiprocessing.set_start_method("spawn", force=True)
logger.info("CUDA available:", torch.cuda.is_available()) logger.info("CUDA available", cuda_available=torch.cuda.is_available())
logger.info("CUDA version PyTorch was built with:", torch.version.cuda) logger.info("CUDA version PyTorch was built with", cuda_version=torch.version.cuda)
async def wait_for_opensearch(): async def wait_for_opensearch():