resolve name conflict

diffusion_sampler.py

@@ -98,10 +98,10 @@ class DiffusionSampler(keras.Model):
         return sqrt_alpha_cum_prod_prev * x0_t + x_t_dir * pred_noise + c1 * random_noise
 
     def noise_scheduler(self, scheduler: str, max_beta: int = 0.02) -> tf.Tensor:
-        pi, t = [tf.constant(num, dtype=tf.float64) for num in (math.pi, self.timesteps)]
+        pi, T = [tf.constant(num, dtype=tf.float64) for num in (math.pi, self.timesteps)]
 
-        alpha_bar = lambda t: tf.math.cos((t + 0.008) / 1.008 * pi / 2) ** 2
-        cosine_scheduler = lambda i: tf.minimum(1 - alpha_bar((i + 1) / t) / alpha_bar(i / t), max_beta)
+        alpha_bar = lambda i: tf.math.cos((i + 0.008) / 1.008 * pi / 2) ** 2
+        cosine_scheduler = lambda t: tf.minimum(1 - alpha_bar((t + 1) / T) / alpha_bar(t / T), max_beta)
 
         if scheduler == "linear":
             x = tf.linspace(start=self.beta_start, stop=self.beta_end, num=self.timesteps)