1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

remove print statements

This commit is contained in:
Aryan
2024-08-21 11:52:05 +02:00
parent 761c44d116
commit 6830fb0805

View File

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