Skip to content

Commit a019867

Browse files
bghirabghirapatrickvonplaten
authored
Remove logger.info statement from Unet2DCondition code to ensure torch compile reliably succeeds (huggingface#4982)
* Remove logger.info statement from Unet2DCondition code to ensure torch compile reliably succeeds * Convert logging statement to a comment for future archaeologists * Update src/diffusers/models/unet_2d_condition.py Co-authored-by: Patrick von Platen <[email protected]> --------- Co-authored-by: bghira <[email protected]> Co-authored-by: Patrick von Platen <[email protected]>
1 parent abc47de commit a019867

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/unet_2d_condition.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -784,7 +784,7 @@ def forward(
784784
upsample_size = None
785785

786786
if any(s % default_overall_up_factor != 0 for s in sample.shape[-2:]):
787-
logger.info("Forward upsample size to force interpolation output size.")
787+
# Forward upsample size to force interpolation output size.
788788
forward_upsample_size = True
789789

790790
# ensure attention_mask is a bias, and give it a singleton query_tokens dimension

0 commit comments

Comments
 (0)