Update app.py

app.py

@@ -29,6 +29,7 @@ spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant
 # Inference function (maps to CLIPTextEncode + EmptySD3LatentImage + KSampler + VAEDecode)
 # ---------------------------------------------------------------------
 @spaces.GPU
+@spaces.GPU
 def generate_images(
     positive: str,
     negative: str,
@@ -49,15 +50,19 @@ def generate_images(
 
     images = []
 
-    # Base seed logic; if seed < 0, random base, else deterministic series
-    if seed < 0:
-        base_seed = torch.randint(0, 2**63 - 1, (1,), device=run_device).item()
-    else:
-        base_seed = int(seed)
+    # seed >= 0  -> deterministic series: seed, seed+1, ...
+    # seed < 0   -> fully random seeds per image
+    fixed_base_seed = int(seed) if seed >= 0 else None
 
     for i in range(num_images):
-        this_seed = base_seed if seed < 0 else base_seed + i
-        generator = torch.Generator(device=run_device).manual_seed(this_seed)
+        if fixed_base_seed is None:
+            # random seed for this image
+            this_seed = torch.randint(0, 2**63 - 1, (1,), device=run_device).item()
+        else:
+            # deterministic offset
+            this_seed = fixed_base_seed + i
+
+        generator = torch.Generator(device=run_device).manual_seed(int(this_seed))
 
         out = pipe(
             prompt=positive,
@@ -66,7 +71,7 @@ def generate_images(
             height=height,
             num_inference_steps=steps,
             guidance_scale=float(cfg),
-            num_images_per_prompt=1,   # always 1 for AOTI safety
+            num_images_per_prompt=1,
             generator=generator,
         ).images[0]
 
@@ -76,6 +81,7 @@ def generate_images(
 
 
 
+
 # ---------------------------------------------------------------------
 # Gradio UI (inputs correspond to Comfy node widgets_values)
 # ---------------------------------------------------------------------