update

a-r-r-o-w · a-r-r-o-w · commit 54b6b3f8be75 · 2024-10-25T08:59:04.000+02:00
diff --git a/tests/models/transformers/test_models_transformer_mochi.py b/tests/models/transformers/test_models_transformer_mochi.py
@@ -18,10 +18,7 @@
 import torch
 
 from diffusers import MochiTransformer3DModel
-from diffusers.utils.testing_utils import (
-    enable_full_determinism,
-    torch_device
-)
+from diffusers.utils.testing_utils import enable_full_determinism, torch_device
 
 from ..test_modeling_common import ModelTesterMixin
 
