diff options
| author | missionfloyd <missionfloyd@users.noreply.github.com> | 2023-03-25 20:51:25 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-25 20:51:25 +0000 |
| commit | 6f18c9b13f06112d6afe9be062fe5308767ea38a (patch) | |
| tree | d82ea08976ca2b5793f449094308aca3c9629099 /modules/models/diffusion/uni_pc/uni_pc.py | |
| parent | 575c17a8f9bc6471a7a0891b665ec42073a18049 (diff) | |
| parent | a0d07fb5807ad55c8ccfdfc9a6d9ae3c62b9d211 (diff) | |
| download | stable-diffusion-webui-gfx803-6f18c9b13f06112d6afe9be062fe5308767ea38a.tar.gz stable-diffusion-webui-gfx803-6f18c9b13f06112d6afe9be062fe5308767ea38a.tar.bz2 stable-diffusion-webui-gfx803-6f18c9b13f06112d6afe9be062fe5308767ea38a.zip | |
Merge branch 'master' into extra-networks-toggle
Diffstat (limited to 'modules/models/diffusion/uni_pc/uni_pc.py')
| -rw-r--r-- | modules/models/diffusion/uni_pc/uni_pc.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/modules/models/diffusion/uni_pc/uni_pc.py b/modules/models/diffusion/uni_pc/uni_pc.py index e9a093a2..eb5f4e76 100644 --- a/modules/models/diffusion/uni_pc/uni_pc.py +++ b/modules/models/diffusion/uni_pc/uni_pc.py @@ -1,6 +1,7 @@ import torch import torch.nn.functional as F import math +from tqdm.auto import trange class NoiseScheduleVP: @@ -750,7 +751,7 @@ class UniPC: if method == 'multistep': assert steps >= order, "UniPC order must be < sampling steps" timesteps = self.get_time_steps(skip_type=skip_type, t_T=t_T, t_0=t_0, N=steps, device=device) - print(f"Running UniPC Sampling with {timesteps.shape[0]} timesteps, order {order}") + #print(f"Running UniPC Sampling with {timesteps.shape[0]} timesteps, order {order}") assert timesteps.shape[0] - 1 == steps with torch.no_grad(): vec_t = timesteps[0].expand((x.shape[0])) @@ -766,7 +767,7 @@ class UniPC: self.after_update(x, model_x) model_prev_list.append(model_x) t_prev_list.append(vec_t) - for step in range(order, steps + 1): + for step in trange(order, steps + 1): vec_t = timesteps[step].expand(x.shape[0]) if lower_order_final: step_order = min(order, steps + 1 - step) |