From 633d890e4964e070be9b0a5b299c2f2e51d4b055 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 17 Oct 2022 12:27:53 +0200 Subject: Upstream updates; better handling of textual embedding --- schedulers/scheduling_euler_a.py | 3 --- 1 file changed, 3 deletions(-) (limited to 'schedulers/scheduling_euler_a.py') 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 -- cgit v1.2.3-54-g00ecf