[example] fix requirements (#2488)

2 years ago · fcc6d61d92
4 changed files with 7 additions and 0 deletions
--- a/examples/language/gpt/experiments/pipeline_parallel/requirements.txt
+++ b/examples/language/gpt/experiments/pipeline_parallel/requirements.txt
@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1
--- a/examples/language/gpt/gemini/requirements.txt
+++ b/examples/language/gpt/gemini/requirements.txt
@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1
--- a/examples/language/gpt/requirements.txt
+++ b/examples/language/gpt/requirements.txt
@ -1 +1,2 @@
 transformers >= 4.23
+colossalai
--- a/examples/language/opt/requirements.txt
+++ b/examples/language/opt/requirements.txt
@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1