change code style

LuChengTHU · LuChengTHU · commit 11960927411e · 2022-11-04T15:21:36.000+08:00
diff --git a/src/diffusers/schedulers/scheduling_dpmsolver_discrete_flax.py b/src/diffusers/schedulers/scheduling_dpmsolver_discrete_flax.py
@@ -23,7 +23,7 @@
 import jax.numpy as jnp
 
 from ..configuration_utils import ConfigMixin, register_to_config
-from .scheduling_utils_flax import FlaxSchedulerMixin, FlaxSchedulerOutput, broadcast_to_shape_from_left
+from .scheduling_utils_flax import FlaxSchedulerMixin, FlaxSchedulerOutput
 
 
 def betas_for_alpha_bar(num_diffusion_timesteps: int, max_beta=0.999) -> jnp.ndarray:
