Oysiyl commited on
Commit
6dcf22b
·
1 Parent(s): a25a409

disable vaetiling looks not so good

Browse files
Files changed (1) hide show
  1. app.py +13 -30
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
- if tile_size is not None:
1157
- # Use tiled decode for larger images
1158
- vaedecode_8 = vaedecodetiled.decode(
1159
- samples=get_value_at_index(ksampler_3, 0),
1160
- vae=get_value_at_index(checkpointloadersimple_4, 2),
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
- if tile_size is not None:
1464
- decoded = vaedecodetiled.decode(
1465
- samples=get_value_at_index(samples, 0),
1466
- vae=get_value_at_index(checkpointloadersimple_artistic, 2),
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)