From b8dd881262e36f69ccb314507db4eb5625403da9 Mon Sep 17 00:00:00 2001
From: Angelo Trotta <111277410+ShyVortex@users.noreply.github.com>
Date: Sat, 29 Mar 2025 17:54:59 +0100
Subject: [PATCH 1/6] Update dependencies

Fix errors when using pip install -r requirements.txt
---
 .idea/inspectionProfiles/Project_Default.xml | 56 ++++++++++++++++++++
 requirements.txt                             | 12 +++--
 2 files changed, 63 insertions(+), 5 deletions(-)

diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml
index ddb2b9e..42fa044 100644
--- a/.idea/inspectionProfiles/Project_Default.xml
+++ b/.idea/inspectionProfiles/Project_Default.xml
@@ -12,6 +12,62 @@
     <inspection_tool class="LoopConditionNotUpdatedInsideLoop" enabled="false" level="WARNING" enabled_by_default="false">
       <option name="ignoreIterators" value="false" />
     </inspection_tool>
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="48">
+            <item index="0" class="java.lang.String" itemvalue="networkx" />
+            <item index="1" class="java.lang.String" itemvalue="nvidia-nvjitlink-cu12" />
+            <item index="2" class="java.lang.String" itemvalue="huggingface-hub" />
+            <item index="3" class="java.lang.String" itemvalue="nvidia-cuda-cupti-cu12" />
+            <item index="4" class="java.lang.String" itemvalue="PyYAML" />
+            <item index="5" class="java.lang.String" itemvalue="nvidia-cufft-cu12" />
+            <item index="6" class="java.lang.String" itemvalue="nvidia-cusparse-cu12" />
+            <item index="7" class="java.lang.String" itemvalue="MarkupSafe" />
+            <item index="8" class="java.lang.String" itemvalue="numpy" />
+            <item index="9" class="java.lang.String" itemvalue="requests" />
+            <item index="10" class="java.lang.String" itemvalue="torchvision" />
+            <item index="11" class="java.lang.String" itemvalue="importlib-metadata" />
+            <item index="12" class="java.lang.String" itemvalue="Jinja2" />
+            <item index="13" class="java.lang.String" itemvalue="nvidia-nccl-cu12" />
+            <item index="14" class="java.lang.String" itemvalue="fsspec" />
+            <item index="15" class="java.lang.String" itemvalue="nvidia-cusolver-cu12" />
+            <item index="16" class="java.lang.String" itemvalue="nvidia-curand-cu12" />
+            <item index="17" class="java.lang.String" itemvalue="filelock" />
+            <item index="18" class="java.lang.String" itemvalue="safetensors" />
+            <item index="19" class="java.lang.String" itemvalue="zipp" />
+            <item index="20" class="java.lang.String" itemvalue="sentencepiece" />
+            <item index="21" class="java.lang.String" itemvalue="certifi" />
+            <item index="22" class="java.lang.String" itemvalue="accelerate" />
+            <item index="23" class="java.lang.String" itemvalue="nvidia-cuda-runtime-cu12" />
+            <item index="24" class="java.lang.String" itemvalue="urllib3" />
+            <item index="25" class="java.lang.String" itemvalue="sympy" />
+            <item index="26" class="java.lang.String" itemvalue="tokenizers" />
+            <item index="27" class="java.lang.String" itemvalue="nvidia-cuda-nvrtc-cu12" />
+            <item index="28" class="java.lang.String" itemvalue="transformers" />
+            <item index="29" class="java.lang.String" itemvalue="triton" />
+            <item index="30" class="java.lang.String" itemvalue="diffusers" />
+            <item index="31" class="java.lang.String" itemvalue="opencv-python" />
+            <item index="32" class="java.lang.String" itemvalue="nvidia-cudnn-cu12" />
+            <item index="33" class="java.lang.String" itemvalue="packaging" />
+            <item index="34" class="java.lang.String" itemvalue="torch" />
+            <item index="35" class="java.lang.String" itemvalue="psutil" />
+            <item index="36" class="java.lang.String" itemvalue="bitsandbytes" />
+            <item index="37" class="java.lang.String" itemvalue="tqdm" />
+            <item index="38" class="java.lang.String" itemvalue="nvidia-cublas-cu12" />
+            <item index="39" class="java.lang.String" itemvalue="regex" />
+            <item index="40" class="java.lang.String" itemvalue="nvidia-nvtx-cu12" />
+            <item index="41" class="java.lang.String" itemvalue="mpmath" />
+            <item index="42" class="java.lang.String" itemvalue="peft" />
+            <item index="43" class="java.lang.String" itemvalue="typing_extensions" />
+            <item index="44" class="java.lang.String" itemvalue="charset-normalizer" />
+            <item index="45" class="java.lang.String" itemvalue="pillow" />
+            <item index="46" class="java.lang.String" itemvalue="GPUtil" />
+            <item index="47" class="java.lang.String" itemvalue="idna" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
     <inspection_tool class="PyUnresolvedReferencesInspection" enabled="false" level="WARNING" enabled_by_default="false">
       <option name="ignoredIdentifiers">
         <list>
diff --git a/requirements.txt b/requirements.txt
index 1ddba46..8d23798 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 accelerate==0.26.1
 certifi==2024.7.4
 charset-normalizer==3.3.2
-diffusers==0.29.2
+diffusers==0.32.2
 filelock==3.13.1
 fsspec==2023.12.2
 huggingface-hub==0.24.3
@@ -16,7 +16,7 @@ nvidia-cublas-cu12==12.1.3.1
 nvidia-cuda-cupti-cu12==12.1.105
 nvidia-cuda-nvrtc-cu12==12.1.105
 nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
+nvidia-cudnn-cu12==9.1.0.70
 nvidia-cufft-cu12==11.0.2.54
 nvidia-curand-cu12==10.3.2.106
 nvidia-cusolver-cu12==11.4.5.107
@@ -34,13 +34,15 @@ regex==2023.12.25
 requests==2.32.2
 safetensors==0.4.1
 sympy==1.12
-tokenizers==0.15.0
+tokenizers==0.21.0
 torch==2.4.0
-torchvision==0.16.2
+torchvision==0.19.0
 tqdm==4.66.3
 transformers==4.48.0
-triton==2.1.0
+triton==3.0.0
 typing_extensions==4.9.0
 urllib3==2.2.2
 zipp==3.19.1
 sentencepiece==0.2.0
+bitsandbytes==0.41.2
+GPUtil==1.4.0

From 6397d8921301d9a7b82f4730f584c033a243f56b Mon Sep 17 00:00:00 2001
From: Angelo Trotta <111277410+ShyVortex@users.noreply.github.com>
Date: Mon, 31 Mar 2025 11:09:30 +0200
Subject: [PATCH 2/6] Implement stable-diffusion-3-5-turbo pipeline

---
 .../it/unimol/diffusiontool/generate_sd2-1.py |   2 +-
 .../it/unimol/diffusiontool/generate_sd3-5.py | 118 ++++++++++++++++++
 .../it/unimol/diffusiontool/generate_sd3.py   |   2 +-
 3 files changed, 120 insertions(+), 2 deletions(-)
 create mode 100644 src/main/python/it/unimol/diffusiontool/generate_sd3-5.py

diff --git a/src/main/python/it/unimol/diffusiontool/generate_sd2-1.py b/src/main/python/it/unimol/diffusiontool/generate_sd2-1.py
index b31095b..c32bc73 100644
--- a/src/main/python/it/unimol/diffusiontool/generate_sd2-1.py
+++ b/src/main/python/it/unimol/diffusiontool/generate_sd2-1.py
@@ -29,7 +29,7 @@ def main():
     # Process the prompt and set the output path
     with torch.cuda.amp.autocast():
         image = pipe(prompt=prompt, negative_prompt=tags, num_inference_steps=25).images[0]
-    output_folder = os.path.abspath("result/generated/sd2-1")
+    output_folder = os.path.abspath("result/generated/general/sd2-1")
     output_filename = f"generated_image_{date}.png"
     output_filepath = os.path.join(output_folder, output_filename)
 
diff --git a/src/main/python/it/unimol/diffusiontool/generate_sd3-5.py b/src/main/python/it/unimol/diffusiontool/generate_sd3-5.py
new file mode 100644
index 0000000..8dc359a
--- /dev/null
+++ b/src/main/python/it/unimol/diffusiontool/generate_sd3-5.py
@@ -0,0 +1,118 @@
+from diffusers import BitsandBytesConfig, SD3TransformerPipeline
+from diffusers import StableDiffusion3Pipeline
+import torch
+import GPUtil
+
+def use_full_model(prompt, tags, date):
+    pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-turbo", torch_dtype=torch.bfloat16)
+    pipe = pipe.to("cuda")
+
+    image = pipe(
+        prompt=prompt,
+        negative_prompt=tags,
+        num_inference_steps=4,
+        guidance_scale=0.0,
+    ).images[0]
+    output_folder = os.path.abspath("result/generated/general/sd3-5")
+    output_filename = f"generated_image_{date}.png"
+    output_filepath = os.path.join(output_folder, output_filename)
+
+    # Check if the output folder exists, and create it if not, then save the image
+    if not os.path.exists(output_folder):
+        os.makedirs(output_folder)
+    image.save(output_filepath)
+
+    # Encode the image as a base64 string
+    with open(output_filepath, "rb") as image_file:
+        encoded_image = base64.b64encode(image_file.read()).decode('utf-8')
+
+    return encoded_image
+
+def use_quantized_model(prompt, tags, date):
+    model_id = "stabilityai/stable-diffusion-3.5-large-turbo"
+
+    nf4_config = BitsAndBytesConfig(
+        load_in_4bit=True,
+        bnb_4bit_quant_type="nf4",
+        bnb_4bit_compute_dtype=torch.bfloat16
+    )
+    model_nf4 = SD3Transformer2DModel.from_pretrained(
+        model_id,
+        subfolder="transformer",
+        quantization_config=nf4_config,
+        torch_dtype=torch.bfloat16
+    )
+
+    t5_nf4 = T5EncoderModel.from_pretrained("diffusers/t5-nf4", torch_dtype=torch.bfloat16)
+
+    pipeline = StableDiffusion3Pipeline.from_pretrained(
+        model_id,
+        transformer=model_nf4,
+        text_encoder_3=t5_nf4,
+        torch_dtype=torch.bfloat16
+    )
+    pipeline.enable_model_cpu_offload()
+
+    image = pipeline(
+        prompt=prompt,
+        negative_prompt=tags,
+        num_inference_steps=4,
+        guidance_scale=0.0,
+        max_sequence_length=512,
+    ).images[0]
+
+    output_folder = os.path.abspath("result/generated/general/sd3-5")
+    output_filename = f"generated_image_{date}.png"
+    output_filepath = os.path.join(output_folder, output_filename)
+
+    # Check if the output folder exists, and create it if not, then save the image
+    if not os.path.exists(output_folder):
+        os.makedirs(output_folder)
+    image.save(output_filepath)
+
+    # Encode the image as a base64 string
+    with open(output_filepath, "rb") as image_file:
+        encoded_image = base64.b64encode(image_file.read()).decode('utf-8')
+
+    return encoded_image
+
+def get_gpu_memory():
+    gpus = GPUtil.getGPUs()
+    all_memory = {gpu.id: gpu.memoryFree for gpu in gpus}
+
+    # Get the GPU with the most free memory
+    gpu_memory = max(all_memory, key=all_memory.get)
+
+    return gpu_memory
+
+def main():
+    # Check if the correct number of command-line arguments is provided
+    if len(sys.argv) != 4:
+        print("Usage: python generate_sd3-5.py <prompt> <tags> <date>")
+        sys.exit(1)
+
+    # Get the prompt and date from the command-line arguments passed from Java
+    prompt = sys.argv[1]
+    tags = sys.argv[2]
+    date = sys.argv[3]
+
+    # Get available video memory and decide which model to use
+    vram = get_gpu_memory()
+    if vram >= 8192:
+        image = use_full_model(
+            prompt=prompt,
+            tags=tags,
+            date=date
+        )
+    else:
+        image = use_quantized_model(
+            prompt=prompt,
+            tags=tags,
+            date=date
+        )
+
+    # Print image as string
+    print(image)
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file
diff --git a/src/main/python/it/unimol/diffusiontool/generate_sd3.py b/src/main/python/it/unimol/diffusiontool/generate_sd3.py
index c676edf..dfa89ba 100644
--- a/src/main/python/it/unimol/diffusiontool/generate_sd3.py
+++ b/src/main/python/it/unimol/diffusiontool/generate_sd3.py
@@ -41,7 +41,7 @@ def main():
             num_inference_steps=25,
             guidance_scale=6.5
         ).images[0]
-    output_folder = os.path.abspath("result/generated/sd3")
+    output_folder = os.path.abspath("result/generated/general/sd3")
     output_filename = f"generated_image_{date}.png"
     output_filepath = os.path.join(output_folder, output_filename)
 

From 03a8d9063ff1af69114e563a4d1834adc24ad02d Mon Sep 17 00:00:00 2001
From: Angelo Trotta <111277410+ShyVortex@users.noreply.github.com>
Date: Mon, 31 Mar 2025 20:46:59 +0200
Subject: [PATCH 3/6] Add scipy, fix bitsandbytes

- Added scipy to requirements.txt
- Updated bitsandbytes to fix deprecated error when running generate_sd3-5.py
---
 .idea/inspectionProfiles/Project_Default.xml | 3 ++-
 requirements.txt                             | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml
index 42fa044..9e18a19 100644
--- a/.idea/inspectionProfiles/Project_Default.xml
+++ b/.idea/inspectionProfiles/Project_Default.xml
@@ -15,7 +15,7 @@
     <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
       <option name="ignoredPackages">
         <value>
-          <list size="48">
+          <list size="49">
             <item index="0" class="java.lang.String" itemvalue="networkx" />
             <item index="1" class="java.lang.String" itemvalue="nvidia-nvjitlink-cu12" />
             <item index="2" class="java.lang.String" itemvalue="huggingface-hub" />
@@ -64,6 +64,7 @@
             <item index="45" class="java.lang.String" itemvalue="pillow" />
             <item index="46" class="java.lang.String" itemvalue="GPUtil" />
             <item index="47" class="java.lang.String" itemvalue="idna" />
+            <item index="48" class="java.lang.String" itemvalue="scipy" />
           </list>
         </value>
       </option>
diff --git a/requirements.txt b/requirements.txt
index 8d23798..58d70c7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -44,5 +44,6 @@ typing_extensions==4.9.0
 urllib3==2.2.2
 zipp==3.19.1
 sentencepiece==0.2.0
-bitsandbytes==0.41.2
+bitsandbytes==0.45.4
 GPUtil==1.4.0
+scipy==1.11.2

From 2c2adc274f1061257613700954b5473ac263d8ea Mon Sep 17 00:00:00 2001
From: Angelo Trotta <111277410+ShyVortex@users.noreply.github.com>
Date: Thu, 17 Apr 2025 10:43:11 +0200
Subject: [PATCH 4/6] Fix model not running

- Added protobuf to requirements.txt
- Added more checks for lack of GPUs
- Added more prints for debug purpose
- Fixed Python output log not showing in DiffusionController
---
 requirements.txt                              |  1 +
 .../controller/DiffusionController.java       |  6 ++-
 .../it/unimol/diffusiontool/generate_sd2-1.py |  2 +-
 .../it/unimol/diffusiontool/generate_sd3-5.py | 47 +++++++++++++++++--
 .../it/unimol/diffusiontool/generate_sd3.py   |  2 +-
 5 files changed, 51 insertions(+), 7 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 58d70c7..23b6fdf 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -47,3 +47,4 @@ sentencepiece==0.2.0
 bitsandbytes==0.45.4
 GPUtil==1.4.0
 scipy==1.11.2
+protobuf==6.30.2
diff --git a/src/main/java/it/unimol/diffusiontool/controller/DiffusionController.java b/src/main/java/it/unimol/diffusiontool/controller/DiffusionController.java
index 634ee32..c5f3948 100644
--- a/src/main/java/it/unimol/diffusiontool/controller/DiffusionController.java
+++ b/src/main/java/it/unimol/diffusiontool/controller/DiffusionController.java
@@ -254,6 +254,7 @@ private void initGenerateView() {
         styleComboBox.getItems().addAll(
                 "Stable Diffusion 2.1",
                 "Stable Diffusion 3",
+                "Stable Diffusion 3.5",
                 "Pixel Art"
         );
         styleComboBox.setPromptText(styleComboBox.getItems().get(0));
@@ -1149,6 +1150,7 @@ public String callPyScript(String prompt, String tags, String date, String path)
             String inputLine;
             while ((inputLine = inputbufferedReader.readLine()) != null) {
                 output.append(inputLine).append("\n");
+                System.out.println(inputLine.trim());
             }
         }
 
@@ -1225,6 +1227,8 @@ public File findPyScript() {
                     fileName = includeUpscaling ? "generate_upscale.py" : "generate_sd2-1.py";
                 else if (styleComboBox.getValue().equals("Stable Diffusion 3"))
                     fileName = "generate_sd3.py";
+                else if (styleComboBox.getValue().equals("Stable Diffusion 3.5"))
+                    fileName = "generate_sd3-5.py";
                 else
                     fileName = "generate_pixart.py";
                 break;
@@ -1343,7 +1347,7 @@ public void throwGenericAlert() {
             genAlert.setHeaderText("ERROR: Upscaling Failure");
             genAlert.setContentText("Something went wrong in the image upscaling. Please retry");
         }
-        genAlert.showAndWait();
+        Platform.runLater(genAlert::showAndWait);
     }
 
     private int checkAvailableSpace() {
diff --git a/src/main/python/it/unimol/diffusiontool/generate_sd2-1.py b/src/main/python/it/unimol/diffusiontool/generate_sd2-1.py
index c32bc73..990b05b 100644
--- a/src/main/python/it/unimol/diffusiontool/generate_sd2-1.py
+++ b/src/main/python/it/unimol/diffusiontool/generate_sd2-1.py
@@ -10,7 +10,7 @@
 def main():
     # Check if the correct number of command-line arguments is provided
     if len(sys.argv) != 4:
-        print("Usage: python generate_sd2-1.py <prompt> <tags> <date>")
+        print("Usage: python generate_sd2-1.py <prompt> <tags> <date>", file=sys.stderr)
         sys.exit(1)
 
     # Get the prompt and date from the command-line arguments passed from Java
diff --git a/src/main/python/it/unimol/diffusiontool/generate_sd3-5.py b/src/main/python/it/unimol/diffusiontool/generate_sd3-5.py
index 8dc359a..c4121bb 100644
--- a/src/main/python/it/unimol/diffusiontool/generate_sd3-5.py
+++ b/src/main/python/it/unimol/diffusiontool/generate_sd3-5.py
@@ -1,18 +1,30 @@
-from diffusers import BitsandBytesConfig, SD3TransformerPipeline
+import sys
+from diffusers import BitsAndBytesConfig, SD3Transformer2DModel
 from diffusers import StableDiffusion3Pipeline
+from transformers import T5EncoderModel
 import torch
 import GPUtil
+import os
+import base64
 
 def use_full_model(prompt, tags, date):
-    pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-turbo", torch_dtype=torch.bfloat16)
+    pipe = StableDiffusion3Pipeline.from_pretrained(
+        "stabilityai/stable-diffusion-3.5-large-turbo",
+        torch_dtype=torch.bfloat16
+    )
     pipe = pipe.to("cuda")
 
+    print("Model loaded successfully. Generating image...", file=sys.stderr)
+
     image = pipe(
         prompt=prompt,
         negative_prompt=tags,
         num_inference_steps=4,
         guidance_scale=0.0,
     ).images[0]
+
+    print("Image generation completed. Saving the image...", file=sys.stderr)
+
     output_folder = os.path.abspath("result/generated/general/sd3-5")
     output_filename = f"generated_image_{date}.png"
     output_filepath = os.path.join(output_folder, output_filename)
@@ -22,6 +34,8 @@ def use_full_model(prompt, tags, date):
         os.makedirs(output_folder)
     image.save(output_filepath)
 
+    print(f"Image saved to {output_filepath}", file=sys.stderr)
+
     # Encode the image as a base64 string
     with open(output_filepath, "rb") as image_file:
         encoded_image = base64.b64encode(image_file.read()).decode('utf-8')
@@ -43,8 +57,12 @@ def use_quantized_model(prompt, tags, date):
         torch_dtype=torch.bfloat16
     )
 
+    print("Quantized model loaded successfully.", file=sys.stderr)
+
     t5_nf4 = T5EncoderModel.from_pretrained("diffusers/t5-nf4", torch_dtype=torch.bfloat16)
 
+    print("T5 Encoder loaded successfully.", file=sys.stderr)
+
     pipeline = StableDiffusion3Pipeline.from_pretrained(
         model_id,
         transformer=model_nf4,
@@ -53,6 +71,8 @@ def use_quantized_model(prompt, tags, date):
     )
     pipeline.enable_model_cpu_offload()
 
+    print("Pipeline initialized. Generating image...", file=sys.stderr)
+
     image = pipeline(
         prompt=prompt,
         negative_prompt=tags,
@@ -61,6 +81,8 @@ def use_quantized_model(prompt, tags, date):
         max_sequence_length=512,
     ).images[0]
 
+    print("Image generation complete. Saving...", file=sys.stderr)
+
     output_folder = os.path.abspath("result/generated/general/sd3-5")
     output_filename = f"generated_image_{date}.png"
     output_filepath = os.path.join(output_folder, output_filename)
@@ -70,6 +92,8 @@ def use_quantized_model(prompt, tags, date):
         os.makedirs(output_folder)
     image.save(output_filepath)
 
+    print(f"Image saved to {output_filepath}", file=sys.stderr)
+
     # Encode the image as a base64 string
     with open(output_filepath, "rb") as image_file:
         encoded_image = base64.b64encode(image_file.read()).decode('utf-8')
@@ -78,17 +102,23 @@ def use_quantized_model(prompt, tags, date):
 
 def get_gpu_memory():
     gpus = GPUtil.getGPUs()
+
+    if not gpus:
+        print("No GPUs detected.", file=sys.stderr)
+        return None
+
     all_memory = {gpu.id: gpu.memoryFree for gpu in gpus}
 
     # Get the GPU with the most free memory
     gpu_memory = max(all_memory, key=all_memory.get)
 
+    print(f"Detected GPU with {all_memory[gpu_memory]} MB free memory.", file=sys.stderr)
     return gpu_memory
 
 def main():
     # Check if the correct number of command-line arguments is provided
     if len(sys.argv) != 4:
-        print("Usage: python generate_sd3-5.py <prompt> <tags> <date>")
+        print("Usage: python generate_sd3-5.py <prompt> <tags> <date>", file=sys.stderr)
         sys.exit(1)
 
     # Get the prompt and date from the command-line arguments passed from Java
@@ -98,13 +128,22 @@ def main():
 
     # Get available video memory and decide which model to use
     vram = get_gpu_memory()
-    if vram >= 8192:
+    if vram is None:
+        print("Running the quantized model...", file=sys.stderr)
+        image = use_quantized_model(
+            prompt=prompt,
+            tags=tags,
+            date=date
+        )
+    elif vram >= 8192:
+        print("Running the full model...", file=sys.stderr)
         image = use_full_model(
             prompt=prompt,
             tags=tags,
             date=date
         )
     else:
+        print("Running the quantized model...", file=sys.stderr)
         image = use_quantized_model(
             prompt=prompt,
             tags=tags,
diff --git a/src/main/python/it/unimol/diffusiontool/generate_sd3.py b/src/main/python/it/unimol/diffusiontool/generate_sd3.py
index dfa89ba..2d0d0d3 100644
--- a/src/main/python/it/unimol/diffusiontool/generate_sd3.py
+++ b/src/main/python/it/unimol/diffusiontool/generate_sd3.py
@@ -10,7 +10,7 @@
 def main():
     # Check if the correct number of command-line arguments is provided
     if len(sys.argv) != 4:
-        print("Usage: python generate_sd3.py <prompt> <tags> <date>")
+        print("Usage: python generate_sd3.py <prompt> <tags> <date>", file=sys.stderr)
         sys.exit(1)
 
     # Get the prompt and date from the command-line arguments passed from Java

From 11e32f487817a97e8e48d53b97a97c0da79ed8f7 Mon Sep 17 00:00:00 2001
From: Angelo Trotta <111277410+ShyVortex@users.noreply.github.com>
Date: Thu, 17 Apr 2025 11:34:00 +0200
Subject: [PATCH 5/6] Bump version to 1.2.0

---
 .../unimol/diffusiontool/application/DiffusionApplication.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/it/unimol/diffusiontool/application/DiffusionApplication.java b/src/main/java/it/unimol/diffusiontool/application/DiffusionApplication.java
index b78074d..c2ae70a 100644
--- a/src/main/java/it/unimol/diffusiontool/application/DiffusionApplication.java
+++ b/src/main/java/it/unimol/diffusiontool/application/DiffusionApplication.java
@@ -55,7 +55,7 @@ public User getUser() {
     }
 
     private void setVersion() {
-        this.version = "1.1.0";
+        this.version = "1.2.0";
     }
 
     public void setRootNode(Parent rootNode) {

From 40619f88cfcef0e64ae27d4c0934f0ce32b69e5c Mon Sep 17 00:00:00 2001
From: Angelo Trotta <111277410+ShyVortex@users.noreply.github.com>
Date: Thu, 17 Apr 2025 11:45:54 +0200
Subject: [PATCH 6/6] Update instructions, usage and credits

---
 README.md | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 1e983bd..a44baa2 100644
--- a/README.md
+++ b/README.md
@@ -17,6 +17,7 @@ The last two are the essential part of the project and they act as GUI for the P
 In order to compile and run the software, it is required that you have the following prerequisites:
 - Open Java Development Kit (OpenJDK) 17 or above
 - Apache Maven (at least version 3.6.3 is recommended)
+- Hugging Face CLI
  
 You also MUST install a Python virtual environment in your home directory, inside a folder named 'venv',
 with the packages listed in *requirements*.
@@ -51,14 +52,20 @@ Then, if you also want a runnable .jar archive, type:
  ```
 With these commands, a new folder named 'target' is created containing the compiled project as well as the executable file.
 
-## Unlock Stable Diffusion 3
-The newest generative model is currently gated, so first you need to sign up [here](https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers).  
+## Usage with Stable Diffusion 3 and 3.5
+The newest generative models are currently gated, so first you need to sign up [here](https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers) 
+and [here](https://huggingface.co/stabilityai/stable-diffusion-3.5-large-turbo).
 Proceed to generate a [token](https://huggingface.co/settings/tokens) under your account settings which you will use to login with:
  ```shell
  huggingface-cli login
  ```
 Enter your credentials first, then the token when it's needed.
 
+It is recommended to pre-download the quantized model of Stable Diffusion 3.5 to avoid long waiting times while using the app:
+ ```shell
+ huggingface-cli download diffusers/t5-nf4
+ ```
+
 ## Screenshots
 ### Home
 ![home-view](https://github.com/user-attachments/assets/50052e5a-c8a4-4eaa-b39f-ae537c81fb9f)
@@ -84,6 +91,7 @@ The project utilizes Stable Diffusion's generative AI pipelines for image genera
 + [stable-diffusion-2-1](https://huggingface.co/stabilityai/stable-diffusion-2-1)
 + [stable-diffusion-2-1-base](https://huggingface.co/stabilityai/stable-diffusion-2-1-base)
 + [stable-diffusion-3-medium](https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers)
++ [stable-diffusion-3.5-large-turbo](https://huggingface.co/stabilityai/stable-diffusion-3.5-large-turbo)
 + [sd-x2-latent-upscaler](https://huggingface.co/stabilityai/sd-x2-latent-upscaler)
 + [pixel-art-style](https://huggingface.co/kohbanye/pixel-art-style)
 + [pixel-art-xl](https://huggingface.co/nerijs/pixel-art-xl)