Skip to content

Commit 6cbb73b

Browse files
committed
fix style
1 parent 95d3a1c commit 6cbb73b

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/diffusers/models/unet_1d.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,6 @@ def forward(
178178
sample = self.final_conv1d_act(sample)
179179
sample = self.final_conv1d_2(sample)
180180

181-
182181
if not return_dict:
183182
return (sample,)
184183

tests/test_models_unet.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515

1616
import gc
1717
import math
18-
import pdb
1918
import tracemalloc
2019
import unittest
2120

@@ -508,7 +507,9 @@ def test_output_pretrained(self):
508507

509508
num_features = model.in_channels
510509
seq_len = 16
511-
noise = torch.randn((1, seq_len, num_features)).permute(0, 2, 1) # match original, we can update values and remove
510+
noise = torch.randn((1, seq_len, num_features)).permute(
511+
0, 2, 1
512+
) # match original, we can update values and remove
512513
time_step = torch.full((num_features,), 0)
513514

514515
with torch.no_grad():

0 commit comments

Comments
 (0)