manual fixes for ruff

This commit is contained in:
AUTOMATIC
2023-05-10 08:25:25 +03:00
parent 762265eab5
commit 96d6ca4199
22 changed files with 129 additions and 129 deletions

View File

@@ -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}")