Spaces:
Running
on
Zero
Running
on
Zero
disable vaetiling looks not so good
Browse files
app.py
CHANGED
|
@@ -1150,23 +1150,14 @@ def _pipeline_standard(
|
|
| 1150 |
log_progress(msg, gr_progress, 0.4)
|
| 1151 |
yield base_qr_pil, msg # Yield with same image as before
|
| 1152 |
|
| 1153 |
-
# Calculate optimal tile size for this image
|
| 1154 |
-
tile_size, overlap = calculate_vae_tile_size(image_size)
|
| 1155 |
|
| 1156 |
-
|
| 1157 |
-
|
| 1158 |
-
|
| 1159 |
-
|
| 1160 |
-
|
| 1161 |
-
tile_size=tile_size,
|
| 1162 |
-
overlap=overlap,
|
| 1163 |
-
)
|
| 1164 |
-
else:
|
| 1165 |
-
# Small image, use standard decode (faster)
|
| 1166 |
-
vaedecode_8 = vaedecode.decode(
|
| 1167 |
-
samples=get_value_at_index(ksampler_3, 0),
|
| 1168 |
-
vae=get_value_at_index(checkpointloadersimple_4, 2),
|
| 1169 |
-
)
|
| 1170 |
|
| 1171 |
# 2) Yield the first decoded image as a second intermediate result
|
| 1172 |
mid_tensor = get_value_at_index(vaedecode_8, 0)
|
|
@@ -1457,21 +1448,13 @@ def _pipeline_artistic(
|
|
| 1457 |
log_progress(msg, gr_progress, 0.4)
|
| 1458 |
yield (noisy_qr_pil if border_size > 0 else base_qr_pil, msg)
|
| 1459 |
|
| 1460 |
-
# First decode with dynamic tiling
|
| 1461 |
-
tile_size, overlap = calculate_vae_tile_size(image_size)
|
| 1462 |
|
| 1463 |
-
|
| 1464 |
-
|
| 1465 |
-
|
| 1466 |
-
|
| 1467 |
-
tile_size=tile_size,
|
| 1468 |
-
overlap=overlap,
|
| 1469 |
-
)
|
| 1470 |
-
else:
|
| 1471 |
-
decoded = vaedecode.decode(
|
| 1472 |
-
samples=get_value_at_index(samples, 0),
|
| 1473 |
-
vae=get_value_at_index(checkpointloadersimple_artistic, 2),
|
| 1474 |
-
)
|
| 1475 |
|
| 1476 |
# Show first pass result
|
| 1477 |
first_pass_tensor = get_value_at_index(decoded, 0)
|
|
|
|
| 1150 |
log_progress(msg, gr_progress, 0.4)
|
| 1151 |
yield base_qr_pil, msg # Yield with same image as before
|
| 1152 |
|
| 1153 |
+
# Calculate optimal tile size for this image - disable for now
|
| 1154 |
+
# tile_size, overlap = calculate_vae_tile_size(image_size)
|
| 1155 |
|
| 1156 |
+
# Small image, use standard decode (faster)
|
| 1157 |
+
vaedecode_8 = vaedecode.decode(
|
| 1158 |
+
samples=get_value_at_index(ksampler_3, 0),
|
| 1159 |
+
vae=get_value_at_index(checkpointloadersimple_4, 2),
|
| 1160 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1161 |
|
| 1162 |
# 2) Yield the first decoded image as a second intermediate result
|
| 1163 |
mid_tensor = get_value_at_index(vaedecode_8, 0)
|
|
|
|
| 1448 |
log_progress(msg, gr_progress, 0.4)
|
| 1449 |
yield (noisy_qr_pil if border_size > 0 else base_qr_pil, msg)
|
| 1450 |
|
| 1451 |
+
# First decode with dynamic tiling - disable for now
|
| 1452 |
+
# tile_size, overlap = calculate_vae_tile_size(image_size)
|
| 1453 |
|
| 1454 |
+
decoded = vaedecode.decode(
|
| 1455 |
+
samples=get_value_at_index(samples, 0),
|
| 1456 |
+
vae=get_value_at_index(checkpointloadersimple_artistic, 2),
|
| 1457 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1458 |
|
| 1459 |
# Show first pass result
|
| 1460 |
first_pass_tensor = get_value_at_index(decoded, 0)
|