Learner commited on
Commit
12b6864
·
1 Parent(s): c67a0fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/learner", from_flax=True, dtype=jnp.float32
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(