From 3de111afbb92c0a958367a5b09b6902c66f17eed Mon Sep 17 00:00:00 2001 From: binmakeswell Date: Tue, 17 Jan 2023 11:01:18 +0800 Subject: [PATCH] [example] fix requirements --- .../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 000000000000..137a69e80498 --- /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 000000000000..137a69e80498 --- /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 e1f131468fb8..ef58bb76bfc8 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 000000000000..137a69e80498 --- /dev/null +++ b/examples/language/opt/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1