add seeding and avoid latest transformers

app.py

@@ -33,6 +33,7 @@ def gen(
     style: str,
     base_prompt: str,
     addon_prompt: str = "",
+    seed: int = -1,
 ):
     global current_dtg_model, current_sdxl_model, sdxl_pipe
     if sdxl_model != current_sdxl_model:
@@ -44,7 +45,9 @@ def gen(
         current_dtg_model = dtg_model
 
     t0 = time_ns()
-    seed = random.randint(0, 2**31 - 1)
+    seed = int(seed)
+    if seed == -1:
+        seed = random.randint(0, 2**31 - 1)
 
     prompt = (
         f"{base_prompt}, {addon_prompt}, "
@@ -79,7 +82,11 @@ def gen(
     torch.cuda.empty_cache()
     t1 = time_ns()
 
-    return result.convert("RGB"), full_prompt, f"Cost: {(t1 - t0) / 1e9:.4}sec"
+    return (
+        result.convert("RGB"),
+        full_prompt,
+        f"Cost: {(t1 - t0) / 1e9:.4}sec || Seed: {seed}",
+    )
 
 
 if __name__ == "__main__":
@@ -119,18 +126,26 @@ click "Next" button until you get the dragon girl you like.
                         label="DTG Model",
                         value=models.model_list[0],
                     )
-                base_prompt = gr.Textbox(
-                    label="Base prompt",
-                    lines=1,
-                    value="1girl, solo, dragon girl, dragon wings, dragon horns, dragon tail",
-                    interactive=False,
-                )
                 with gr.Row():
+                    base_prompt = gr.Textbox(
+                        label="Base prompt",
+                        lines=1,
+                        value="1girl, solo, dragon girl, dragon wings, dragon horns, dragon tail",
+                        interactive=False,
+                    )
                     addon_propmt = gr.Textbox(
                         label="Addon prompt",
                         lines=1,
                         value="cowboy shot",
                     )
+                with gr.Row():
+                    seed = gr.Number(
+                        label="Seed (-1 for random)",
+                        value=-1,
+                        minimum=-1,
+                        maximum=2**31 - 1,
+                        precision=0,
+                    )
                     style = gr.Dropdown(
                         DEFAULT_STYLE_LIST,
                         label="Style",
@@ -146,7 +161,7 @@ click "Next" button until you get the dragon girl you like.
 
         submit.click(
             fn=gen,
-            inputs=[sdxl_model, dtg_model, style, base_prompt, addon_propmt],
+            inputs=[sdxl_model, dtg_model, style, base_prompt, addon_propmt, seed],
             outputs=[result, dtg_output, cost_time],
         )
 

diff.py

@@ -23,6 +23,7 @@ def set_timesteps_polyexponential(self, orig_sigmas, num_inference_steps, device
 
 def model_forward(k_diffusion_model: torch.nn.Module):
     orig_forward = k_diffusion_model.forward
+
     def forward(*args, **kwargs):
         with torch.autocast(device_type="cuda", dtype=torch.float16):
             result = orig_forward(*args, **kwargs)

requirements.txt

@@ -1,5 +1,5 @@
 diffusers
-transformers
+transformers<4.39
 k_diffusion
 requests
 sentencepiece