fix: Only install stanford-stk on linux (#169)

predibase · Jan 9, 2024 · b14207c · b14207c
1 parent 112aeec
commit b14207c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/server/poetry.lock b/server/poetry.lock
diff --git a/server/pyproject.toml b/server/pyproject.toml
@@ -36,7 +36,7 @@ torch = {version = "2.1.1", optional = true }
 peft = {version = "0.4.0", optional = true }
 boto3 = "^1.28.34"
 urllib3 = "<=1.26.18"
-stanford-stk = "^0.0.6"
+stanford-stk = { version = "^0.0.6", markers = "sys_platform == 'linux'" }
 
 [tool.poetry.extras]
 torch = ["torch"]

diff --git a/server/requirements.txt b/server/requirements.txt
@@ -43,12 +43,12 @@ safetensors==0.3.1 ; python_version >= "3.9" and python_version < "4.0"
 sentencepiece==0.1.99 ; python_version >= "3.9" and python_version < "4.0"
 setuptools==69.0.2 ; python_version >= "3.9" and python_version < "4.0"
 six==1.16.0 ; python_version >= "3.9" and python_version < "4.0"
-stanford-stk==0.0.6 ; python_version >= "3.9" and python_version < "4.0"
+stanford-stk==0.0.6 ; python_version >= "3.9" and python_version < "4.0" and sys_platform == "linux"
 tiktoken==0.5.2 ; python_version >= "3.9" and python_version < "4.0"
 tokenizers==0.15.0 ; python_version >= "3.9" and python_version < "4.0"
 tqdm==4.66.1 ; python_version >= "3.9" and python_version < "4.0"
 transformers==4.36.0 ; python_version >= "3.9" and python_version < "4.0"
-triton==2.1.0 ; python_version >= "3.9" and python_version < "4.0"
+triton==2.1.0 ; python_version >= "3.9" and python_version < "4.0" and sys_platform == "linux"
 typer==0.6.1 ; python_version >= "3.9" and python_version < "4.0"
 typing-extensions==4.9.0 ; python_version >= "3.9" and python_version < "4.0"
 urllib3==1.26.18 ; python_version >= "3.9" and python_version < "4.0"