Cognomen commited on
Commit
bd11ec1
Β·
1 Parent(s): 97416e9

disable unipc and enable DPM solver

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -33,9 +33,10 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
33
  "./models/wd-1-5-b2-flax",
34
  controlnet=cnet,
35
  revision="flax",
36
- dtype=jnp.bfloat16
 
37
  )
38
- pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
39
  #pipe.enable_model_cpu_offload()
40
  #pipe.enable_xformers_memory_efficient_attention()
41
 
 
33
  "./models/wd-1-5-b2-flax",
34
  controlnet=cnet,
35
  revision="flax",
36
+ dtype=jnp.bfloat16,
37
+ scheduler="FlaxDPMSolverMultistepScheduler"
38
  )
39
+ #pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
40
  #pipe.enable_model_cpu_offload()
41
  #pipe.enable_xformers_memory_efficient_attention()
42