Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove Unnecessary Line (Scheduler)

This commit is contained in:
cafeai 2022-11-06 19:34:33 +09:00
parent b303fdc293
commit 2048ce67ef
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
diffusers_trainer.py
View File

@ -714,7 +714,6 @@ def main():
beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", skip_prk_steps=True 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( pipeline = StableDiffusionPipeline(
text_encoder=text_encoder, text_encoder=text_encoder,
vae=vae, vae=vae,