Merge pull request #35 from Shreyanand/gpu-footprint

Gpu footprint
redhat-et · May 3, 2023 · a55d14a · a55d14a
2 parents 72c3d48 + aeec314
commit a55d14a
Show file tree

Hide file tree

Showing 4 changed files with 1,637 additions and 817 deletions.
diff --git a/Pipfile b/Pipfile
@@ -9,16 +9,24 @@ coverage = "*"
 Sphinx = "*"
 
 [packages]
+ipywidgets = "*"
+ipynb = "*"
+boto3 = "*"
+python-dotenv = "*"
+langchain = "*"
+transformers = "*"
+torch = "*"
+unstructured = "*"
+GPUtil = "*" 
+numba = "*"
 awscli = "*"
 pypdf2 = "*"
 nltk = "*"
 rouge = "*"
 jiwer = "*"
 evaluate = "*"
-langchain = "*"
 openai = "*"
 chromadb = "*"
-unstructured = "*"
 bert-score = "*"
 
 [requires]