removed print statement

This commit is contained in:
Yousef Rafat 2025-12-23 12:36:10 +02:00
parent 5b0c80a093
commit d41b1111eb

View File

@ -962,10 +962,6 @@ class UNetMidBlock3D(nn.Module):
attentions = []
if attention_head_dim is None:
print(
f"It is not recommend to pass `attention_head_dim=None`. "
f"Defaulting `attention_head_dim` to `in_channels`: {in_channels}."
)
attention_head_dim = in_channels
for _ in range(num_layers):