diff --git a/examples/images/diffusion/environment.yaml b/examples/images/diffusion/environment.yaml index 5b5579211..69904c72e 100644 --- a/examples/images/diffusion/environment.yaml +++ b/examples/images/diffusion/environment.yaml @@ -27,4 +27,5 @@ dependencies: - torchmetrics==0.7.0 - prefetch_generator - datasets + - colossalai - -e . diff --git a/examples/images/diffusion/requirements.txt b/examples/images/diffusion/requirements.txt index 333f32d6e..60c4b903e 100644 --- a/examples/images/diffusion/requirements.txt +++ b/examples/images/diffusion/requirements.txt @@ -14,4 +14,5 @@ webdataset==0.2.5 open-clip-torch==2.7.0 gradio==3.11 datasets +colossalai -e . diff --git a/examples/images/dreambooth/requirements.txt b/examples/images/dreambooth/requirements.txt index 1ec828c63..6c4f40fb5 100644 --- a/examples/images/dreambooth/requirements.txt +++ b/examples/images/dreambooth/requirements.txt @@ -5,3 +5,4 @@ transformers>=4.21.0 ftfy tensorboard modelcards +colossalai diff --git a/examples/images/vit/requirements.txt b/examples/images/vit/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/images/vit/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/language/palm/requirements.txt b/examples/language/palm/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/language/palm/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/language/roberta/requirements.txt b/examples/language/roberta/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/language/roberta/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/tutorial/auto_parallel/requirements.txt b/examples/tutorial/auto_parallel/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/tutorial/auto_parallel/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/tutorial/hybrid_parallel/requirements.txt b/examples/tutorial/hybrid_parallel/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/tutorial/hybrid_parallel/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/tutorial/large_batch_optimizer/requirements.txt b/examples/tutorial/large_batch_optimizer/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/tutorial/large_batch_optimizer/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/tutorial/opt/inference/requirements.txt b/examples/tutorial/opt/inference/requirements.txt index d0970d587..e6e8511e3 100644 --- a/examples/tutorial/opt/inference/requirements.txt +++ b/examples/tutorial/opt/inference/requirements.txt @@ -6,3 +6,4 @@ sanic_ext==22.9.0 torch>=1.10.0 transformers==4.23.1 uvicorn==0.19.0 +colossalai diff --git a/examples/tutorial/requirements.txt b/examples/tutorial/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/tutorial/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/tutorial/sequence_parallel/requirements.txt b/examples/tutorial/sequence_parallel/requirements.txt new file mode 100644 index 000000000..137a69e80 --- /dev/null +++ b/examples/tutorial/sequence_parallel/requirements.txt @@ -0,0 +1,2 @@ +colossalai >= 0.1.12 +torch >= 1.8.1 diff --git a/examples/tutorial/stable_diffusion/environment.yaml b/examples/tutorial/stable_diffusion/environment.yaml index 59baa3c76..7d8aec86f 100644 --- a/examples/tutorial/stable_diffusion/environment.yaml +++ b/examples/tutorial/stable_diffusion/environment.yaml @@ -28,6 +28,7 @@ dependencies: - torchmetrics==0.7.0 - kornia==0.6 - prefetch_generator + - colossalai - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers - -e git+https://github.com/openai/CLIP.git@main#egg=clip - -e . diff --git a/examples/tutorial/stable_diffusion/requirements.txt b/examples/tutorial/stable_diffusion/requirements.txt index 54bc00029..a57003562 100644 --- a/examples/tutorial/stable_diffusion/requirements.txt +++ b/examples/tutorial/stable_diffusion/requirements.txt @@ -16,6 +16,7 @@ torchmetrics==0.6.0 kornia==0.6 opencv-python==4.6.0.66 prefetch_generator +colossalai -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers -e git+https://github.com/openai/CLIP.git@main#egg=clip -e .