extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-28 18:08:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

Typos and minor fixes for sd 2

Browse Source
This commit is contained in:
cmdr2 2022-11-29 13:30:08 +05:30
parent 5432297691
commit ac605e9352
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ui/sd_internal/runtime.py
View File

@ -796,7 +796,7 @@ def _txt2img(opt_W, opt_H, opt_n_samples, opt_ddim_steps, opt_scale, start_code,
if sampler_name == 'ddim': if sampler_name == 'ddim':
thread_data.model.make_schedule(ddim_num_steps=opt_ddim_steps, ddim_eta=opt_ddim_eta, verbose=False) thread_data.model.make_schedule(ddim_num_steps=opt_ddim_steps, ddim_eta=opt_ddim_eta, verbose=False)
samples_ddim, intermediates = thread_data.model.sample( samples_ddim = thread_data.model.sample(
S=opt_ddim_steps, S=opt_ddim_steps,
conditioning=c, conditioning=c,
seed=opt_seed, seed=opt_seed,
@ -837,7 +837,7 @@ def _img2img(init_latent, t_enc, batch_size, opt_scale, c, uc, opt_ddim_steps, o
) )
# decode it # decode it
samples_ddim, intermediates = thread_data.model.sample( samples_ddim = thread_data.model.sample(
t_enc, t_enc,
c, c,
z_enc, z_enc,

2
ui/sd_internal/task_manager.py
View File

@ -290,6 +290,8 @@ def thread_render(device):
current_vae_path = task.request.use_vae_model current_vae_path = task.request.use_vae_model
def step_callback(): def step_callback():
global current_state_error
if isinstance(current_state_error, SystemExit) or isinstance(current_state_error, StopAsyncIteration) or isinstance(task.error, StopAsyncIteration): if isinstance(current_state_error, SystemExit) or isinstance(current_state_error, StopAsyncIteration) or isinstance(task.error, StopAsyncIteration):
runtime.thread_data.stop_processing = True runtime.thread_data.stop_processing = True
if isinstance(current_state_error, StopAsyncIteration): if isinstance(current_state_error, StopAsyncIteration):