fix alignment

diffusion_sampler.py

@@ -32,8 +32,7 @@ class DiffusionSampler(keras.Model):
     ):
         super().__init__(**kwargs)
         self.noise_predictor = load_model(filepath=model, safe_mode=False) if isinstance(model, str) else model
-        self.ema_noise_predictor = load_model(filepath=ema_model, safe_mode=False) if isinstance(model,
-                                                                                                 str) else ema_model
+        self.ema_noise_predictor = load_model(filepath=ema_model, safe_mode=False) if isinstance(model, str) else ema_model
         self.ema = ema
         self.beta_start = beta_start
         self.beta_end = beta_end