[update] add temperature

app.py

@@ -26,7 +26,7 @@ def load_pipeline():
 
 # @spaces.GPU #[uncomment to use ZeroGPU]
 @spaces.GPU(duration=40)
-def generate(prompt, content_type="image", guidance_scale=3.0, seed=42, width=1024, height=1024):
+def generate(prompt, content_type="image", guidance_scale=3.0, temperature=1.0, seed=42, width=1024, height=1024):
     global pipe
     if pipe is None:
         try:
@@ -40,6 +40,7 @@ def generate(prompt, content_type="image", guidance_scale=3.0, seed=42, width=10
         result = pipe(
             prompt=prompt,
             guidance_scale=guidance_scale,
+            temperature=temperature,
             content_type=content_type,
             seed=seed,
             width=width,
@@ -77,6 +78,7 @@ def create_demo():
                 
                 with gr.Accordion("Settings", open=True):
                     guidance_scale = gr.Slider(minimum=0, maximum=10, value=3.0, label="Guidance Scale")
+                    temperature = gr.Slider(minimum=0.1, maximum=1.5, value=1.0, label="Temperature")
                     seed = gr.Number(value=42, label="Seed", precision=0)
                     width = gr.Number(value=1024, label="Width", precision=0)
                     height = gr.Number(value=1024, label="Height", precision=0)
@@ -86,24 +88,24 @@ def create_demo():
             with gr.Column():
                 output = gr.Gallery(label="Output", show_label=True, elem_id="gallery", columns=1, height="auto", preview=True, object_fit="contain")
         
-        def generate_and_display(prompt, content_type, guidance_scale, seed, width, height):
-            result = generate(prompt, content_type, guidance_scale, seed, width, height)
+        def generate_and_display(prompt, content_type, guidance_scale, temperature, seed, width, height):
+            result = generate(prompt, content_type, guidance_scale, temperature, seed, width, height)
             if result:
                 return [result]
             return []
         
         generate_btn.click(
             fn=generate_and_display,
-            inputs=[prompt_input, content_type, guidance_scale, seed, width, height],
+            inputs=[prompt_input, content_type, guidance_scale, temperature, seed, width, height],
             outputs=output
         )
         
         gr.Examples(
             examples=[
-                ["A majestic lion standing on a cliff at sunset", "image", 3.0, 42, 1024, 1024],
-                ["A dog chasing a butterfly in a meadow", "video", 3.0, 123, 1024, 1024],
+                ["A majestic lion standing on a cliff at sunset", "image", 3.0, 1.0, 42, 1024, 1024],
+                ["A dog chasing a butterfly in a meadow", "video", 3.0, 1.0, 123, 1024, 1024],
             ],
-            inputs=[prompt_input, content_type, guidance_scale, seed, width, height],
+            inputs=[prompt_input, content_type, guidance_scale, temperature, seed, width, height],
             cache_examples=False
         )
     

grn_pipeline.py

@@ -155,6 +155,7 @@ class GRNPipeline:
         prompt,
         negative_prompt='',
         guidance_scale=3.0,
+        temperature=1.0,
         num_inference_steps=50,
         width=512,
         height=512,
@@ -169,6 +170,7 @@ class GRNPipeline:
             self.args.seed = np.random.randint(0, 10000)
         
         self.args.cfg_val = guidance_scale
+        self.args.tau = temperature
         
         # Get dynamic resolution meta
         dynamic_resolution_h_w, h_div_w_templates = get_dynamic_resolution_meta(