fiewolf1000 commited on
Commit
1426e3d
·
verified ·
1 Parent(s): aa269af

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
  fastapi>=0.110.0
2
  uvicorn>=0.29.0
3
  pydantic>=2.0.0
4
- transformers>=4.41.0
5
- bitsandbytes>=0.43.0
6
- torch>=2.2.0
 
 
1
  fastapi>=0.110.0
2
  uvicorn>=0.29.0
3
  pydantic>=2.0.0
4
+ transformers>=4.41.0 # 需 4.41+ 版本才能支持 Qwen-0.5B
5
+ bitsandbytes>=0.43.0 # 4bit 量化依赖
6
+ torch>=2.2.0 # 需 2.2+ 版本支持 bfloat16
7
+ accelerate>=0.30.0 # 辅助模型加载,避免显存碎片