(self,
given_betas=None, beta_schedule="linear", timesteps=1000,
linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3)
| 612 | print("### USING STD-RESCALING ###") |
| 613 | |
| 614 | def register_schedule(self, |
| 615 | given_betas=None, beta_schedule="linear", timesteps=1000, |
| 616 | linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3): |
| 617 | super().register_schedule(given_betas, beta_schedule, timesteps, linear_start, linear_end, cosine_s) |
| 618 | |
| 619 | self.shorten_cond_schedule = self.num_timesteps_cond > 1 |
| 620 | if self.shorten_cond_schedule: |
| 621 | self.make_cond_schedule() |
| 622 | |
| 623 | def instantiate_first_stage(self, config): |
| 624 | model = instantiate_from_config(config) |
nothing calls this directly
no test coverage detected