|
@@ -413,7 +413,7 @@ RAG_EMBEDDING_MODEL_AUTO_UPDATE = (
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
-# device type embbeding models - "cpu" (default), "cuda" (nvidia gpu required) or "mps" (apple silicon) - choosing this right can lead to better performance
|
|
|
|
|
|
|
+# 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")
|
|
USE_CUDA = os.environ.get("USE_CUDA_DOCKER", "false")
|
|
|
|
|
|
|
|
if USE_CUDA.lower() == "true":
|
|
if USE_CUDA.lower() == "true":
|
Timothy J. Baek