diff --git a/cldm/ddim_hacked.py b/cldm/ddim_hacked.py index 25b1bc9472..ca460c3099 100644 --- a/cldm/ddim_hacked.py +++ b/cldm/ddim_hacked.py @@ -88,7 +88,7 @@ def sample(self, elif isinstance(conditioning, list): for ctmp in conditioning: if ctmp.shape[0] != batch_size: - print(f"Warning: Got {cbs} conditionings but batch-size is {batch_size}") + print(f"Warning: Got {ctmp.shape[0]} conditionings but batch-size is {batch_size}") else: if conditioning.shape[0] != batch_size: diff --git a/ldm/models/diffusion/ddim.py b/ldm/models/diffusion/ddim.py index 27ead0ea91..454c22929d 100644 --- a/ldm/models/diffusion/ddim.py +++ b/ldm/models/diffusion/ddim.py @@ -88,7 +88,7 @@ def sample(self, elif isinstance(conditioning, list): for ctmp in conditioning: if ctmp.shape[0] != batch_size: - print(f"Warning: Got {cbs} conditionings but batch-size is {batch_size}") + print(f"Warning: Got {ctmp.shape[0]} conditionings but batch-size is {batch_size}") else: if conditioning.shape[0] != batch_size: