r3gm commited on
Commit
6d92970
1 Parent(s): 66ab4b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -22,7 +22,7 @@ else:
22
  pipe = pipe.to(device)
23
 
24
  MAX_SEED = np.iinfo(np.int32).max
25
- MAX_IMAGE_SIZE = 1024 + 512
26
 
27
  def latents_to_rgb(latents):
28
  weights = (
@@ -48,7 +48,7 @@ class BaseGenerator:
48
  self.image = None
49
  self.new_image_event = threading.Event()
50
  self.generation_finished = threading.Event()
51
- self.intermediate_image_concurrency(5)
52
 
53
  def intermediate_image_concurrency(self, concurrency):
54
  self.concurrency = concurrency
@@ -139,7 +139,9 @@ with gr.Blocks(css=css) as demo:
139
  with gr.Column(elem_id="col-container"):
140
  gr.Markdown(f"""
141
  # Text-to-Image: Display each generation step
 
142
  Gradio template for displaying preview images during generation steps
 
143
  Currently running on {power_device}.
144
  """)
145
 
 
22
  pipe = pipe.to(device)
23
 
24
  MAX_SEED = np.iinfo(np.int32).max
25
+ MAX_IMAGE_SIZE = 1536
26
 
27
  def latents_to_rgb(latents):
28
  weights = (
 
48
  self.image = None
49
  self.new_image_event = threading.Event()
50
  self.generation_finished = threading.Event()
51
+ self.intermediate_image_concurrency(3)
52
 
53
  def intermediate_image_concurrency(self, concurrency):
54
  self.concurrency = concurrency
 
139
  with gr.Column(elem_id="col-container"):
140
  gr.Markdown(f"""
141
  # Text-to-Image: Display each generation step
142
+
143
  Gradio template for displaying preview images during generation steps
144
+
145
  Currently running on {power_device}.
146
  """)
147