This commit is contained in:
Timothy J. Baek
2024-09-19 20:56:13 +02:00
parent b0bc36f2af
commit 2dad9b9432
3 changed files with 16 additions and 1 deletions

View File

@@ -31,6 +31,7 @@ try:
except ImportError:
print("dotenv not installed, skipping...")
DOCKER = os.environ.get("DOCKER", "False").lower() == "true"
# device type embedding models - "cpu" (default), "cuda" (nvidia gpu required) or "mps" (apple silicon) - choosing this right can lead to better performance
USE_CUDA = os.environ.get("USE_CUDA_DOCKER", "false")