Skip to content

Commit

Permalink
Pin minimum transformers to 4.39 to reduce Llama/Gemma memory pressure
Browse files Browse the repository at this point in the history
  • Loading branch information
arnavgarg1 committed Mar 21, 2024
1 parent 9eba97e commit f0575a2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ torchaudio
torchtext
torchvision
pydantic<2.0
transformers>=4.38.1
transformers>=4.39.0
tifffile
imagecodecs
tokenizers>=0.15
Expand Down

0 comments on commit f0575a2

Please sign in to comment.