diff --git a/diffusers/scripts/exp_ldm_sd_gsa.sh b/diffusers/scripts/exp_ldm_sd_gsa.sh
index 6b8c0b1..fcb5d8c 100755
--- 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
\ No newline at end of file
+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
\ No newline at end of file
diff --git a/diffusers/scripts/exp_sdxl_demo.sh b/diffusers/scripts/exp_sdxl_demo.sh
index a2e476a..394238d 100755
--- 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
index f55cca5..9eec747 100644
--- 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)
\ No newline at end of file
+        return GSAStableDiffusionPipelineOutput(images=None, gsa_features=gsa_features)
\ No newline at end of file