diff --git a/requirements.txt b/examples/causal_language_modeling/requirements.txt similarity index 64% rename from requirements.txt rename to examples/causal_language_modeling/requirements.txt index 10f15c0..5e53d13 100644 --- a/requirements.txt +++ b/examples/causal_language_modeling/requirements.txt @@ -2,3 +2,6 @@ transformers accelerate loralib evaluate +deepspeed +tqdm +datasets \ No newline at end of file diff --git a/examples/conditional_generation/requirements.txt b/examples/conditional_generation/requirements.txt new file mode 100644 index 0000000..5e53d13 --- /dev/null +++ b/examples/conditional_generation/requirements.txt @@ -0,0 +1,7 @@ +transformers +accelerate +loralib +evaluate +deepspeed +tqdm +datasets \ No newline at end of file diff --git a/examples/lora_dreambooth/requirements.txt b/examples/lora_dreambooth/requirements.txt new file mode 100644 index 0000000..f1163a4 --- /dev/null +++ b/examples/lora_dreambooth/requirements.txt @@ -0,0 +1,11 @@ +transformers +accelerate +loralib +evaluate +deepspeed +tqdm +datasets +diffusers +Pillow +torchvision +huggingface_hub \ No newline at end of file diff --git a/examples/sequence_classification/requirements.txt b/examples/sequence_classification/requirements.txt new file mode 100644 index 0000000..5e53d13 --- /dev/null +++ b/examples/sequence_classification/requirements.txt @@ -0,0 +1,7 @@ +transformers +accelerate +loralib +evaluate +deepspeed +tqdm +datasets \ No newline at end of file diff --git a/examples/token_classification/requirements.txt b/examples/token_classification/requirements.txt new file mode 100644 index 0000000..4d010ee --- /dev/null +++ b/examples/token_classification/requirements.txt @@ -0,0 +1,9 @@ +transformers +accelerate +loralib +evaluate +deepspeed +tqdm +datasets +Pillow +torchvision \ No newline at end of file