Spaces:

xiaopeng
/

Z_Image_Turbo_Free

Running on Zero

simapeng commited on Jan 22

Commit

9b941df

1 Parent(s): ee7ac8f

6th

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import io
 import tempfile
 import zipfile
 import torch
 import spaces
 import gradio as gr
@@ -271,7 +272,7 @@ def generate_image(
 ):
     """Generate N images using a deterministic seed cascade (x1..xN)."""
     if randomize_seed:
-        seed = torch.randint(0, MAX_SEED, (1,)).item()
     base_seed = int(seed) % MAX_SEED
     if base_seed < 0:

 import tempfile
 import zipfile
+import random
 import torch
 import spaces
 import gradio as gr
 ):
     """Generate N images using a deterministic seed cascade (x1..xN)."""
     if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
     base_seed = int(seed) % MAX_SEED
     if base_seed < 0: