Merge pull request #7 from moreh-dev/thuchk/add-req-file

Thuchk/add req file
moreh-dev · Sep 6, 2023 · 27643d5 · 27643d5
2 parents 44732bf + 7fbb88b
commit 27643d5
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -183,4 +183,5 @@ dreambooth-model/
 instruct-pix2pix-model/
 text-inversion-model/
 text-to-image-model/
-unconditional-image-generation-model/
+unconditional-image-generation-model/
+outputs/
diff --git a/examples/requirements.txt b/examples/requirements.txt
@@ -0,0 +1,11 @@
+-e ../
+git+https://github.com/huggingface/[email protected]
+torchvision>=0.11.0
+protobuf>=3.19.6,<4
+transformers>=4.25.1
+ftfy==6.1.1
+tensorboard==2.13.0
+Jinja2==3.1.2
+matplotlib==3.7.2
+opencv-python>=4.8.0
+datasets==2.14.2
diff --git a/examples/unconditional_image_generation/train_unconditional_mlflow.py b/examples/unconditional_image_generation/train_unconditional_mlflow.py
@@ -311,7 +311,6 @@ def main(args):
         gradient_accumulation_steps=args.gradient_accumulation_steps,
         mixed_precision=args.mixed_precision,
         log_with=args.logger,
-        logging_dir=logging_dir,
         project_dir=logging_dir,
         project_config=accelerator_project_config,
     )