Remove Unnecessary Line (Scheduler)

2022-11-06 19:34:33 +09:00 · 2022-11-06 19:34:33 +09:00 · 2048ce67ef
parent b303fdc293
commit 2048ce67ef
1 changed files with 0 additions and 1 deletions
--- a/diffusers_trainer.py
+++ b/diffusers_trainer.py
@ -714,7 +714,6 @@ def main():
                                beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", skip_prk_steps=True
                            )

-                        scheduler.set_timesteps(num_inference_steps=35)
                        pipeline = StableDiffusionPipeline(
                            text_encoder=text_encoder,
                            vae=vae,