From 6830fb08052039c549708f4a8ec77d3d4c29b6d7 Mon Sep 17 00:00:00 2001 From: Aryan Date: Wed, 21 Aug 2024 11:52:05 +0200 Subject: [PATCH] remove print statements --- src/diffusers/pipelines/free_noise_utils.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/src/diffusers/pipelines/free_noise_utils.py b/src/diffusers/pipelines/free_noise_utils.py index 817360b3ea..1c1147a494 100644 --- a/src/diffusers/pipelines/free_noise_utils.py +++ b/src/diffusers/pipelines/free_noise_utils.py @@ -68,12 +68,6 @@ class ChunkedInferenceModule(nn.Module): inputs = dict(zip(chunked_inputs.keys(), chunked_input)) inputs.update(kwargs) - for key, input in inputs.items(): - if torch.is_tensor(input): - print(key, input.shape) - else: - print(key) - intermediate_tensor_or_tensor_tuple = self.module(*args, **inputs) results.append(intermediate_tensor_or_tensor_tuple)