Hugging Face's logo

Spaces:
WiNE-iNEFF
/
Mineskin-Diffusion
Running

App Files Community
1
WiNE-iNEFF commited on
Commit
70c0578
·
1 Parent(s): 7040426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -18,7 +18,10 @@ device = (
18
 
19
  pipeline_name = 'WiNE-iNEFF/Minecraft-Skin-Diffusion-V3'
20
  image_pipe = DDPMPipeline.from_pretrained(pipeline_name, use_safetensors=True).to(device)
21
- scheduler = DDIMScheduler.from_pretrained(pipeline_name)
 
 
 
22
  scheduler.set_timesteps(num_inference_steps=25)
23
 
24
  def show_images_save(x):
 
18
 
19
  pipeline_name = 'WiNE-iNEFF/Minecraft-Skin-Diffusion-V3'
20
  image_pipe = DDPMPipeline.from_pretrained(pipeline_name, use_safetensors=True).to(device)
21
+ try:
22
+ scheduler = DDIMScheduler.from_pretrained(pipeline_name)
23
+ except:
24
+ scheduler = DDIMScheduler.from_pretrained(pipeline_name, subfolder="scheduler")
25
  scheduler.set_timesteps(num_inference_steps=25)
26
 
27
  def show_images_save(x):