Add 6 new samplers; Fix a bug where new tasks wouldn't started if a previous task was stopped

This commit is contained in:
cmdr2 2022-12-17 16:45:43 +05:30
parent 8189b38e6e
commit 1595f1ed05
2 changed files with 15 additions and 8 deletions

View File

@ -133,14 +133,20 @@
</td></tr> </td></tr>
<tr id="samplerSelection" class="pl-5"><td><label for="sampler_name">Sampler:</label></td><td> <tr id="samplerSelection" class="pl-5"><td><label for="sampler_name">Sampler:</label></td><td>
<select id="sampler_name" name="sampler_name"> <select id="sampler_name" name="sampler_name">
<option value="plms">plms</option> <option value="plms">PLMS</option>
<option value="ddim">ddim</option> <option value="ddim">DDIM</option>
<option value="heun">heun</option> <option value="heun">Heun</option>
<option value="euler">euler</option> <option value="euler">Euler</option>
<option value="euler_a" selected>euler_a</option> <option value="euler_a" selected>Euler Ancestral</option>
<option value="dpm2">dpm2</option> <option value="dpm2">DPM2</option>
<option value="dpm2_a">dpm2_a</option> <option value="dpm2_a">DPM2 Ancestral</option>
<option value="lms">lms</option> <option value="lms">LMS</option>
<option value="dpm_solver_stability">DPM Solver (Stability AI)</option>
<option value="dpmpp_2s_a" selected>DPM++ 2s Ancestral</option>
<option value="dpmpp_2m">DPM++ 2m</option>
<option value="dpmpp_sde">DPM++ SDE</option>
<option value="dpm_fast">DPM Fast</option>
<option value="dpm_adaptive">DPM Adaptive</option>
</select> </select>
<a href="https://github.com/cmdr2/stable-diffusion-ui/wiki/How-to-Use#samplers" target="_blank"><i class="fa-solid fa-circle-question help-btn"><span class="simple-tooltip top-left">Click to learn more about samplers</span></i></a> <a href="https://github.com/cmdr2/stable-diffusion-ui/wiki/How-to-Use#samplers" target="_blank"><i class="fa-solid fa-circle-question help-btn"><span class="simple-tooltip top-left">Click to learn more about samplers</span></i></a>
</td></tr> </td></tr>

View File

@ -27,6 +27,7 @@ def init(device):
device_manager.device_init(context, device) device_manager.device_init(context, device)
def make_images(req: GenerateImageRequest, task_data: TaskData, data_queue: queue.Queue, task_temp_images: list, step_callback): def make_images(req: GenerateImageRequest, task_data: TaskData, data_queue: queue.Queue, task_temp_images: list, step_callback):
context.stop_processing = False
log.info(f'request: {save_utils.get_printable_request(req)}') log.info(f'request: {save_utils.get_printable_request(req)}')
log.info(f'task data: {task_data.dict()}') log.info(f'task data: {task_data.dict()}')