From 2048ce67ef5a9c73749c034439ea9dd78de32372 Mon Sep 17 00:00:00 2001 From: cafeai <116491182+cafeai@users.noreply.github.com> Date: Sun, 6 Nov 2022 19:34:33 +0900 Subject: [PATCH] Remove Unnecessary Line (Scheduler) --- diffusers_trainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/diffusers_trainer.py b/diffusers_trainer.py index eb364f8..719078a 100644 --- 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,