Merge pull request #386 from anhuong/v2.1.1

chore(release): merge set of changes for v2.1.1
foundation-model-stack · Nov 6, 2024 · e2ac091 · e2ac091
2 parents 8f16818 + 0e664f2
commit e2ac091
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -132,7 +132,7 @@ Example: Train.jsonl
 
 ## Supported Models
 
-Current supported and tested models are `Llama2` (7 and 13B configurations have been tested) and `GPTBigCode`.
+Current supported and tested models are `Llama3` (8B configuration has been tested) and `GPTBigCode`.
 
 ## Training
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -28,7 +28,7 @@ classifiers=[
 dependencies = [
 "numpy>=1.26.4,<2.0",
 "accelerate>=0.20.3,!=0.34,<1.1",
-"transformers>4.41,<4.50",
+"transformers>4.41,<4.46",
 "torch>=2.2.0,<2.5",
 "sentencepiece>=0.1.99,<0.3",
 "tokenizers>=0.13.3,<1.0",