add requirements

21d4e4cb · MPU王荣胜 · GitHub · 3f17f2b7 · 21d4e4cb · 21d4e4cb
Unverified Commit 21d4e4cb authored May 25, 2023 by MPU王荣胜 Committed by GitHub May 25, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 16 additions and 0 deletions

requirements.txt requirements.txt +6 -0

requirements_wo_ds.txt requirements_wo_ds.txt +10 -0

No files found.
--- a/requirements.txt
+++ b/requirements.txt
+SwissArmyTransformer>=0.3.6
+torch>1.10.0
+torchvision
+transformers>=4.27.1
+mdtex2html
+gradio
\ No newline at end of file
--- a/requirements_wo_ds.txt
+++ b/requirements_wo_ds.txt
+torch>1.10.0
+torchvision
+transformers>=4.27.1
+mdtex2html
+gradio
+sentencepiece
+tensorboardX
+datasets
+cpm_kernels
+einops
\ No newline at end of file