[SD] Removed img2dataset installation from Docker to the download script

mlcommons · Sep 7, 2023 · d74e2d4 · d74e2d4
1 parent b9bdb31
commit d74e2d4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/stable_diffusion/requirements.txt b/stable_diffusion/requirements.txt
@@ -19,7 +19,6 @@ datasets==2.10.1
 colossalai==0.2.7
 invisible-watermark==0.1.5
 diffusers==0.14.0
-img2dataset==1.41.0
 cloudpathlib==0.13.0
 git+https://github.com/facebookresearch/xformers.git@5eb0dbf315d14b5f7b38ac2ff3d8379beca7df9b#egg=xformers
 bitsandbytes==0.37.2

diff --git a/stable_diffusion/scripts/datasets/laion400m-download-dataset.sh b/stable_diffusion/scripts/datasets/laion400m-download-dataset.sh
@@ -25,6 +25,7 @@ done
 
 mkdir -p ${OUTPUT_DIR}
 
+pip install img2dataset==1.41.0
 img2dataset \
   --url_list ${METADATA_DIR} \
   --input_format "parquet" \