Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
|
38 |
"jax-diffusers-event/canny-coyo1m", dtype=jnp.float32
|
39 |
)
|
40 |
pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
41 |
-
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, dtype=jnp.float32
|
42 |
)
|
43 |
|
44 |
def infer(prompts, negative_prompts, image):
|
|
|
38 |
"jax-diffusers-event/canny-coyo1m", dtype=jnp.float32
|
39 |
)
|
40 |
pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
41 |
+
"runwayml/stable-diffusion-v1-5", from_pt=True, controlnet=controlnet, dtype=jnp.float32
|
42 |
)
|
43 |
|
44 |
def infer(prompts, negative_prompts, image):
|