Merge pull request #697 from ruivieira/RHOAIENG-12195-trustyai

RHOAIENG-12195: Update TrustyAI dependencies
opendatahub-io · Sep 18, 2024 · d4183a4 · d4183a4
2 parents 83ccdb5 + 16efc38
commit d4183a4
Show file tree

Hide file tree

Showing 4 changed files with 2,738 additions and 1,307 deletions.
diff --git a/jupyter/trustyai/ubi9-python-3.11/Pipfile b/jupyter/trustyai/ubi9-python-3.11/Pipfile
@@ -3,12 +3,21 @@ url = "https://pypi.org/simple"
 verify_ssl = true
 name = "pypi"
 
+[[source]]
+url = "https://download.pytorch.org/whl/cu121"
+verify_ssl = true
+name = "pytorch"
+
 [dev-packages]
 
 [packages]
+# PyTorch packages
+torch = {version = "~=2.2.1", index = "pytorch"}
 # TrustyAI packages
-trustyai = ">=0.6.0"
-
+transformers = "~=4.36.2"
+datasets = "~=2.21.0"
+accelerate = "~=0.34.2"
+trustyai = "~=0.6.1"
 # Datascience and useful extensions
 boto3 = "~=1.35.14"
 kafka-python = "~=2.0.2"