fix style

huggingface · Oct 12, 2022 · 6cbb73b · 6cbb73b
1 parent 95d3a1c
commit 6cbb73b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/diffusers/models/unet_1d.py b/src/diffusers/models/unet_1d.py
@@ -178,7 +178,6 @@ def forward(
  sample = self.final_conv1d_act(sample)
  sample = self.final_conv1d_2(sample)
 
-
  if not return_dict:
  return (sample,)
 

diff --git a/tests/test_models_unet.py b/tests/test_models_unet.py
@@ -15,7 +15,6 @@
 
 import gc
 import math
-import pdb
 import tracemalloc
 import unittest
 
@@ -508,7 +507,9 @@ def test_output_pretrained(self):
 
  num_features = model.in_channels
  seq_len = 16
- noise = torch.randn((1, seq_len, num_features)).permute(0, 2, 1) # match original, we can update values and remove
+ noise = torch.randn((1, seq_len, num_features)).permute(
+ 0, 2, 1
+ ) # match original, we can update values and remove
  time_step = torch.full((num_features,), 0)
 
  with torch.no_grad():