lower batchsize

caradryanl · May 23, 2024 · ff47a18 · ff47a18
1 parent 022237c
commit ff47a18
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/diffusers/scripts/exp_ldm_sd_gsa.sh b/diffusers/scripts/exp_ldm_sd_gsa.sh
@@ -1,4 +1,4 @@
-python scripts/train_gsa.py --model-type sd --gsa-mode 1 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --batch-size 3
-python scripts/train_gsa.py --model-type sd --gsa-mode 2 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/  --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --batch-size 3
-python scripts/train_gsa.py --model-type ldm --gsa-mode 1 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 3
-python scripts/train_gsa.py --model-type ldm --gsa-mode 2 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 3
+python scripts/train_gsa.py --model-type sd --gsa-mode 1 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --batch-size 4
+python scripts/train_gsa.py --model-type sd --gsa-mode 2 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/  --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --batch-size 4
+python scripts/train_gsa.py --model-type ldm --gsa-mode 1 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 2
+python scripts/train_gsa.py --model-type ldm --gsa-mode 2 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 2
diff --git a/diffusers/scripts/exp_sdxl_demo.sh b/diffusers/scripts/exp_sdxl_demo.sh
@@ -1 +1 @@
-python scripts/train_secmi.py --model-type sdxl --ckpt-path ../models/diffusers/Kohaku-XL-Epsilon/ --member-dataset hakubooru-2-5k-member --holdout-dataset hakubooru-2-5k-nonmember --batch-size 2 --demo True
+python scripts/train_gsa.py --model-type sdxl --gsa-mode 1 --ckpt-path ../models/diffusers/Kohaku-XL-Epsilon/ --member-dataset hakubooru-2-5k-member --holdout-dataset hakubooru-2-5k-nonmember --batch-size 1 --demo True
diff --git a/diffusers/stable_copyright/gsa_pipeline_sdxl.py b/diffusers/stable_copyright/gsa_pipeline_sdxl.py
@@ -11,7 +11,8 @@
         randn_tensor,
     )
 
-from .secmi_pipeline_stable_diffusion import SecMIStableDiffusionPipelineOutput
+from .gsa_pipeline_latent_diffusion import GSAStableDiffusionPipelineOutput
+
 
 class GSAStableDiffusionXLPipeline(
     StableDiffusionXLPipeline
@@ -397,4 +398,4 @@ def __call__(
         if not return_dict:
             return (image,)
 
-        return SecMIStableDiffusionPipelineOutput(images=image, denoising_results=denoising_results, posterior_results=posterior_results)
+        return GSAStableDiffusionPipelineOutput(images=None, gsa_features=gsa_features)
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		python scripts/train_secmi.py --model-type sdxl --ckpt-path ../models/diffusers/Kohaku-XL-Epsilon/ --member-dataset hakubooru-2-5k-member --holdout-dataset hakubooru-2-5k-nonmember --batch-size 2 --demo True
		python scripts/train_gsa.py --model-type sdxl --gsa-mode 1 --ckpt-path ../models/diffusers/Kohaku-XL-Epsilon/ --member-dataset hakubooru-2-5k-member --holdout-dataset hakubooru-2-5k-nonmember --batch-size 1 --demo True