Updates requirements

- Updates requirements.txt iree to enable larger hal allocations - Adds requirements.txt for stateless_llama.py and gen_external_params
nod-ai · Nov 17, 2023 · 0694bef · 0694bef
1 parent e5ae42f
commit 0694bef
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/python/turbine_models/custom_models/requirements.txt b/python/turbine_models/custom_models/requirements.txt
@@ -0,0 +1,3 @@
+transformers
+sentencepiece
+protobuf
diff --git a/python/turbine_models/gen_external_params/requirements.txt b/python/turbine_models/gen_external_params/requirements.txt
@@ -0,0 +1,2 @@
+brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b
+transformers
diff --git a/requirements.txt b/requirements.txt
@@ -7,5 +7,5 @@
 -r pytorch-cpu-requirements.txt
 -r torchvision-requirements.txt
 
-iree-compiler==20231113.707
-iree-runtime==20231113.707
+iree-compiler==20231117.711
+iree-runtime==20231117.711
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b
		transformers