Spaces:

Cognomen
/

CatCon-Controlnet-WD-1-5-b2

Runtime error

Cognomen commited on May 3, 2023

Commit

26eb517

1 Parent(s): 3d5deda

fix scheduler

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,9 +34,8 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         controlnet=cnet,
         revision="flax",
         dtype=jnp.bfloat16,
-        scheduler=FlaxScoreSdeVeScheduler()
         )
-#pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()
 #pipe.enable_xformers_memory_efficient_attention()

         controlnet=cnet,
         revision="flax",
         dtype=jnp.bfloat16,
         )
+pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()
 #pipe.enable_xformers_memory_efficient_attention()