Spaces:

WiNE-iNEFF
/

Mineskin-Diffusion

Sleeping

WiNE-iNEFF commited on Sep 9, 2023

Commit

63f8629

1 Parent(s): 88972dd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ device = (
 pipeline_name = 'WiNE-iNEFF/Minecraft-Skin-Diffusion-V2'
 image_pipe = DDPMPipeline.from_pretrained(pipeline_name).to(device)
 scheduler = DDIMScheduler.from_pretrained(pipeline_name)
-scheduler.set_timesteps(num_inference_steps=20)
 def show_images_save(x):
     grid = torchvision.utils.make_grid(x, nrow=4, padding=0)

 pipeline_name = 'WiNE-iNEFF/Minecraft-Skin-Diffusion-V2'
 image_pipe = DDPMPipeline.from_pretrained(pipeline_name).to(device)
 scheduler = DDIMScheduler.from_pretrained(pipeline_name)
+scheduler.set_timesteps(num_inference_steps=25)
 def show_images_save(x):
     grid = torchvision.utils.make_grid(x, nrow=4, padding=0)