Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def image_grid(imgs, rows, cols):
|
|
43 |
|
44 |
# load control net and stable diffusion v1-5
|
45 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
46 |
-
"jax-diffuser-event
|
47 |
)
|
48 |
|
49 |
pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
|
|
43 |
|
44 |
# load control net and stable diffusion v1-5
|
45 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
46 |
+
"learner/jax-diffuser-event", from_flax=True, dtype=jnp.float32
|
47 |
)
|
48 |
|
49 |
pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|