ford442 commited on
Commit
3719417
·
verified ·
1 Parent(s): d3850c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -263,7 +263,7 @@ def generate_30(
263
  "num_inference_steps": num_inference_steps,
264
  "generator": generator,
265
  "output_type": "pil",
266
- "callback_on_step_end": scheduler_swap_callback,
267
  }
268
  if use_resolution_binning:
269
  options["use_resolution_binning"] = True
@@ -322,7 +322,7 @@ def generate_60(
322
  "num_inference_steps": num_inference_steps,
323
  "generator": generator,
324
  "output_type": "pil",
325
- "callback_on_step_end": scheduler_swap_callback,
326
  }
327
  if use_resolution_binning:
328
  options["use_resolution_binning"] = True
@@ -380,7 +380,7 @@ def generate_90(
380
  "num_inference_steps": num_inference_steps,
381
  "generator": generator,
382
  "output_type": "pil",
383
- "callback_on_step_end": scheduler_swap_callback,
384
  }
385
  if use_resolution_binning:
386
  options["use_resolution_binning"] = True
 
263
  "num_inference_steps": num_inference_steps,
264
  "generator": generator,
265
  "output_type": "pil",
266
+ # "callback_on_step_end": scheduler_swap_callback,
267
  }
268
  if use_resolution_binning:
269
  options["use_resolution_binning"] = True
 
322
  "num_inference_steps": num_inference_steps,
323
  "generator": generator,
324
  "output_type": "pil",
325
+ # "callback_on_step_end": scheduler_swap_callback,
326
  }
327
  if use_resolution_binning:
328
  options["use_resolution_binning"] = True
 
380
  "num_inference_steps": num_inference_steps,
381
  "generator": generator,
382
  "output_type": "pil",
383
+ # "callback_on_step_end": scheduler_swap_callback,
384
  }
385
  if use_resolution_binning:
386
  options["use_resolution_binning"] = True