Oysiyl commited on
Commit
89a59b3
·
1 Parent(s): 72ccbb3

Fix warmup parameters: use module_size=12 and error_correction='Medium (15%)' to match generate_all_triton_kernels.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -427,8 +427,8 @@ def compile_models_with_aoti():
427
  input_type="URL",
428
  image_size=512,
429
  border_size=0,
430
- error_correction="M",
431
- module_size=1,
432
  module_drawer="Square",
433
  seed=TEST_SEED,
434
  enable_upscale=False,
@@ -460,8 +460,8 @@ def compile_models_with_aoti():
460
  input_type="URL",
461
  image_size=640,
462
  border_size=0,
463
- error_correction="M",
464
- module_size=1,
465
  module_drawer="Square",
466
  seed=TEST_SEED,
467
  enable_upscale=False,
@@ -509,8 +509,8 @@ def compile_models_with_aoti():
509
  input_type="URL",
510
  image_size=512,
511
  border_size=0,
512
- error_correction="M",
513
- module_size=1,
514
  module_drawer="Square",
515
  seed=TEST_SEED,
516
  enable_upscale=False,
@@ -527,8 +527,8 @@ def compile_models_with_aoti():
527
  input_type="URL",
528
  image_size=640,
529
  border_size=0,
530
- error_correction="M",
531
- module_size=1,
532
  module_drawer="Square",
533
  seed=TEST_SEED,
534
  enable_upscale=False,
 
427
  input_type="URL",
428
  image_size=512,
429
  border_size=0,
430
+ error_correction="Medium (15%)",
431
+ module_size=12,
432
  module_drawer="Square",
433
  seed=TEST_SEED,
434
  enable_upscale=False,
 
460
  input_type="URL",
461
  image_size=640,
462
  border_size=0,
463
+ error_correction="Medium (15%)",
464
+ module_size=12,
465
  module_drawer="Square",
466
  seed=TEST_SEED,
467
  enable_upscale=False,
 
509
  input_type="URL",
510
  image_size=512,
511
  border_size=0,
512
+ error_correction="Medium (15%)",
513
+ module_size=12,
514
  module_drawer="Square",
515
  seed=TEST_SEED,
516
  enable_upscale=False,
 
527
  input_type="URL",
528
  image_size=640,
529
  border_size=0,
530
+ error_correction="Medium (15%)",
531
+ module_size=12,
532
  module_drawer="Square",
533
  seed=TEST_SEED,
534
  enable_upscale=False,