Upload New File

56c0adc3 · wanglch · 5462ff1c · 56c0adc3
Commit 56c0adc3 authored May 23, 2024 by wanglch
Hide whitespace changes
Inline Side-by-side

Showing with 23 additions and 0 deletions

docker/requirements.txt docker/requirements.txt +23 -0

No files found.
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
+torch
+transformers==4.37.2
+bitsandbytes==0.43.0
+accelerate
+colorama
+cpm_kernels
+sentencepiece
+streamlit
+transformers_stream_generator
+peft==0.8.2
+trl>=0.7.6
+gradio>=3.38.0,<4.0.0
+scipy
+einops
+sentencepiece
+protobuf
+jieba
+rouge-chinese
+nltk
+uvicorn
+pydantic
+fastapi
+sse-starlette
\ No newline at end of file