glm4/basic_demo/requirements.txt

23 lines
580 B
Plaintext
Raw Normal View History

2024-08-11 17:59:00 +08:00
torch>=2.4.0
torchvision>=0.19.0
transformers==4.44.0
huggingface-hub>=0.24.5
2024-06-05 10:22:16 +08:00
sentencepiece>=0.2.0
2024-07-24 10:17:07 +08:00
jinja2>=3.1.4
2024-07-16 17:08:50 +08:00
pydantic>=2.8.2
2024-08-11 17:59:00 +08:00
timm>=1.0.8
2024-06-05 10:22:16 +08:00
tiktoken>=0.7.0
2024-08-11 17:59:00 +08:00
numpy==1.26.4 # Need less than 2.0.0
accelerate>=0.33.0
2024-07-16 17:08:50 +08:00
sentence_transformers>=3.0.1
2024-09-06 15:14:25 +08:00
gradio>=4.42.0 # web demo
openai>=1.43.0 # openai demo
2024-07-16 17:08:50 +08:00
einops>=0.8.0
2024-07-24 10:17:07 +08:00
pillow>=10.4.0
2024-08-11 17:59:00 +08:00
sse-starlette>=2.1.3
bitsandbytes>=0.43.3 # INT4 Loading
2024-06-06 16:18:14 +08:00
2024-08-11 17:59:00 +08:00
# vllm==0.5.4 # using with VLLM Framework
# flash-attn>=2.6.1 # using with flash-attention 2
2024-06-06 16:18:14 +08:00
# PEFT model, not need if you don't use PEFT finetune model.
2024-08-11 17:59:00 +08:00
# peft>=0.12.2 # Using with finetune model