diff --git a/image_generation/Storyteller-Studio-Filter/requirements.txt b/image_generation/Storyteller-Studio-Filter/requirements.txt
index d8e0755..6bbbc52 100644
--- a/image_generation/Storyteller-Studio-Filter/requirements.txt
+++ b/image_generation/Storyteller-Studio-Filter/requirements.txt
@@ -1,30 +1,48 @@
+accelerate==0.26.1
+antlr4-python3-runtime==4.9.3
 certifi==2022.12.7
 charset-normalizer==2.1.1
+controlnet-aux==0.0.3
+diffusers==0.23.1
+einops==0.7.0
 filelock==3.9.0
 fsspec==2023.10.0
 huggingface-hub==0.18.0
+icefall==1.0
 idna==3.4
+imageio==2.33.1
+importlib-metadata==7.0.1
 Jinja2==3.1.2
+lazy_loader==0.3
 MarkupSafe==2.1.2
 mpmath==1.3.0
 networkx==3.0
 numpy==1.24.1
+omegaconf==2.3.0
+opencv-python==4.9.0.80
 packaging==23.2
 Pillow==9.3.0
+psutil==5.9.7
 PyYAML==6.0.1
 regex==2023.10.3
 requests==2.28.1
 ruff==0.1.3
 safetensors==0.4.0
+scikit-image==0.22.0
+scipy==1.11.4
 sympy==1.12
+tifffile==2023.12.9
+timm==0.9.12
 tokenizers==0.13.3
 torch==2.1.0+cu118
 torchaudio==2.1.0+cu118
+torchsde==0.2.6
 torchvision==0.16.0+cu118
 tqdm==4.66.1
+trampoline==0.1.2
 transformers==4.30.2
 triton==2.1.0
 typing_extensions==4.4.0
 urllib3==1.26.13
 xformers==0.0.22.post7+cu118
-diffusers
+zipp==3.17.0
diff --git a/image_generation/Storyteller-Studio-Filter/test.sh b/image_generation/Storyteller-Studio-Filter/test.sh
index 0268912..6f297dc 100755
--- a/image_generation/Storyteller-Studio-Filter/test.sh
+++ b/image_generation/Storyteller-Studio-Filter/test.sh
@@ -1,14 +1,15 @@
 python3.10 main.py \
---prompt "This is a prompt" \
+--prompt "raiden mei, pantyhose, boots, single glove, earrings, dress, choker, tank top" \
 --negative-prompt "This is a negative prompt" \
---number-of-samples 20 \
---samplers "DPM++ 2M Karras" \
---width 512 \
---height 256 \
---cfg-scale 10 \
---seed 1234 \
---loRA-path "/path/to/loRA" \
---check-point "/path/to/checkpoint" \
---vae "/path/to/vae" \
+--number-of-samples 128 \
+--samplers "DPM++ 2M SDE Heun Karras" \
+--width 1024 \
+--height 1024 \
+--cfg-scale 8 \
+--seed 1 \
+--loRA-path "xs4vow.safetensors" \
+--check-point "waifuDiffusionBeta03_beta3.safetensors" \
+--vae "vaeFtEma560000Ema_original.pt" \
 --batch-size 1 \
---batch-count 1
+--batch-count 10
+