Skip to content

Commit

Permalink
Update requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
huyiwen committed Apr 16, 2024
1 parent 72986a3 commit 06959de
Showing 1 changed file with 16 additions and 8 deletions.
24 changes: 16 additions & 8 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,22 +2,30 @@ torch
transformers
safetensors
tokenizers
openai==0.28.1
tqdm>=4.58.0
tiktoken>=0.5.0
datasets>=2.16.1
accelerate
coloredlogs
tqdm>=4.58.0

# Efficient inference
packaging
vllm
flash-attn # https://github.com/Dao-AILab/flash-attention/issues/453#issuecomment-1692867770

# API Models
anthropic
dashscope
qianfan
openai==0.28.1
tiktoken>=0.5.0

# Metrics
nltk
sacrebleu
rouge_score
vllm
flash-attn
langcodes
language_data
anthropic
google-api-python-client
immutabledict
langdetect
dashscope
qianfan

0 comments on commit 06959de

Please sign in to comment.