From 393d470044545e4b97a831a2a39520601ff5a3f9 Mon Sep 17 00:00:00 2001 From: George Petterson Date: Thu, 10 Aug 2023 13:36:03 -0400 Subject: [PATCH] Fix langchain requirements pulling in unwanted Cuda packages --- apps/language_models/langchain/langchain_requirements.txt | 3 +++ 1 file changed, 3 insertions(+) diff --git a/apps/language_models/langchain/langchain_requirements.txt b/apps/language_models/langchain/langchain_requirements.txt index 78bd6e7562..8f1fc8df4f 100644 --- a/apps/language_models/langchain/langchain_requirements.txt +++ b/apps/language_models/langchain/langchain_requirements.txt @@ -1,3 +1,6 @@ +-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html +--pre + # for generate (gradio server) and finetune datasets==2.13.0 sentencepiece==0.1.99