1 files changed, 0 insertions, 3 deletions

diff --git a/schedulers/scheduling_euler_a.py b/schedulers/scheduling_euler_a.py
index 6abe971..c097a8a 100644
--- a/schedulers/scheduling_euler_a.py
+++ b/schedulers/scheduling_euler_a.py
@@ -47,7 +47,6 @@ class EulerAScheduler(SchedulerMixin, ConfigMixin):
         beta_end: float = 0.02,
         beta_schedule: str = "linear",
         trained_betas: Optional[np.ndarray] = None,
-        tensor_format: str = "pt",
         num_inference_steps=None,
         device='cuda'
     ):
@@ -63,7 +62,6 @@ class EulerAScheduler(SchedulerMixin, ConfigMixin):
             raise NotImplementedError(f"{beta_schedule} does is not implemented for {self.__class__}")
 
         self.device = device
-        self.tensor_format = tensor_format
 
         self.alphas = 1.0 - self.betas
         self.alphas_cumprod = torch.cumprod(self.alphas, dim=0)
@@ -77,7 +75,6 @@ class EulerAScheduler(SchedulerMixin, ConfigMixin):
         # get sigmas
         self.DSsigmas = ((1 - self.alphas_cumprod) / self.alphas_cumprod) ** 0.5
         self.sigmas = self.get_sigmas(self.DSsigmas, self.num_inference_steps)
-        self.set_format(tensor_format=tensor_format)
 
     # A# take number of steps as input
     # A# store 1) number of steps 2) timesteps 3) schedule