Update app.py
Browse files
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 =
|
| 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(
|
| 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 |
|