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:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user