Cognomen commited on
Commit
ce08539
Β·
1 Parent(s): c465829

add unipc sampler

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -25,6 +25,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
25
  controlnet=cnet,
26
  torch_dtype=torch.float32,
27
  )
 
28
 
29
  generator = torch.manual_seed(0)
30
 
 
25
  controlnet=cnet,
26
  torch_dtype=torch.float32,
27
  )
28
+ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
29
 
30
  generator = torch.manual_seed(0)
31