From fcc6d61d925a6ad3e95e8d71e8f16361595a725f Mon Sep 17 00:00:00 2001 From: binmakeswell Date: Tue, 17 Jan 2023 13:07:25 +0800 Subject: [PATCH] [example] fix requirements (#2488) --- .../language/gpt/experiments/pipeline_parallel/requirements.txt | 2 ++ examples/language/gpt/gemini/requirements.txt | 2 ++ examples/language/gpt/requirements.txt | 1 + examples/language/opt/requirements.txt | 2 ++ 4 files changed, 7 insertions(+) create mode 100644 examples/language/gpt/experiments/pipeline_parallel/requirements.txt create mode 100644 examples/language/gpt/gemini/requirements.txt create mode 100644 examples/language/opt/requirements.txt diff --git a/examples/language/gpt/experiments/pipeline_parallel/requirements.txt b/examples/language/gpt/experiments/pipeline_parallel/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/language/gpt/experiments/pipeline_parallel/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/language/gpt/gemini/requirements.txt b/examples/language/gpt/gemini/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/language/gpt/gemini/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/language/gpt/requirements.txt b/examples/language/gpt/requirements.txt index e1f131468..ef58bb76b 100644 --- a/examples/language/gpt/requirements.txt +++ b/examples/language/gpt/requirements.txt @@ -1 +1,2 @@ transformers >= 4.23 +colossalai diff --git a/examples/language/opt/requirements.txt b/examples/language/opt/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/language/opt/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1