Spaces:

Cognomen
/

CatCon-Controlnet-WD-1-5-b2

Runtime error

Cognomen commited on May 2, 2023

Commit

23ace94

1 Parent(s): 1da736d

load WD with from_pt=True

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,6 +30,7 @@ pipe = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         "./models/wd-1-5-b2",
         controlnet=cnet,
         dtype=jnp.bfloat16,
         )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()

         "./models/wd-1-5-b2",
         controlnet=cnet,
         dtype=jnp.bfloat16,
+        from_pt=True
         )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()