Skip to content

Commit

Permalink
Updated transformers and mpt_7b versions (#31)
Browse files Browse the repository at this point in the history
  • Loading branch information
saileshd1402 authored Dec 13, 2023
1 parent cab08d0 commit 15debb4
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion llm/model_config.json
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
"mpt_7b": {
"handler": "handler.py",
"repo_id": "mosaicml/mpt-7b",
"repo_version": "00f72b21dd089db80c7fb50cb606996751500f6d",
"repo_version": "ada218f9a93b5f1c6dce48a4cc9ff01fcba431e7",
"model_params": {
"temperature": 0.7,
"repetition_penalty": 1.75,
Expand Down
4 changes: 2 additions & 2 deletions llm/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@ torch==2.0.1
torchvision==0.15.2
torchtext==0.15.2
fastai==2.7.12
tokenizers==0.13.3
tokenizers==0.15.0
torchdata==0.6.1
transformers== 4.33.0
transformers== 4.36.0
accelerate==0.22.0
nvgpu==0.10.0
torchserve==0.8.2
Expand Down

0 comments on commit 15debb4

Please sign in to comment.