aboutsummaryrefslogtreecommitdiff
path: root/modules/models/diffusion/uni_pc
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-05-10 08:25:25 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-05-10 08:25:25 +0300
commit96d6ca4199e7c5eee8d451618de5161cea317c40 (patch)
tree8f101a345bcd1d66f4047b5e20918e2058e4dc7c /modules/models/diffusion/uni_pc
parent762265eab58cdb8f2d6398769bab43d8b8db0075 (diff)
manual fixes for ruff
Diffstat (limited to 'modules/models/diffusion/uni_pc')
-rw-r--r--modules/models/diffusion/uni_pc/sampler.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/models/diffusion/uni_pc/sampler.py b/modules/models/diffusion/uni_pc/sampler.py
index a241c8a7..0a9defa1 100644
--- a/modules/models/diffusion/uni_pc/sampler.py
+++ b/modules/models/diffusion/uni_pc/sampler.py
@@ -54,7 +54,8 @@ class UniPCSampler(object):
if conditioning is not None:
if isinstance(conditioning, dict):
ctmp = conditioning[list(conditioning.keys())[0]]
- while isinstance(ctmp, list): ctmp = ctmp[0]
+ while isinstance(ctmp, list):
+ ctmp = ctmp[0]
cbs = ctmp.shape[0]
if cbs != batch_size:
print(f"Warning: Got {cbs} conditionings but batch-size is {batch_size}")