Spaces:

Cognomen
/

CatCon-Controlnet-WD-1-5-b2

Runtime error

Cognomen commited on May 2, 2023

Commit

d3fc59d

1 Parent(s): c848fb3

comma

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ cnet, cnet_params = FlaxControlNetModel.from_pretrained("./models/catcon-control
 pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         "./models/wd-1-5-b2",
         controlnet=cnet,
-        revision="flax"
         dtype=jnp.bfloat16,
         )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)

 pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         "./models/wd-1-5-b2",
         controlnet=cnet,
+        revision="flax",
         dtype=jnp.bfloat16,
         )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)