Update main.py

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

View file

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