bguisard commited on
Commit
2b02469
1 Parent(s): 6484989

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -9,6 +9,7 @@ DTYPE = jnp.float16
9
 
10
  pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
11
  "bguisard/stable-diffusion-nano-2-1",
 
12
  dtype=DTYPE,
13
  )
14
  if DTYPE != jnp.float32:
 
9
 
10
  pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
11
  "bguisard/stable-diffusion-nano-2-1",
12
+ revisio="flax",
13
  dtype=DTYPE,
14
  )
15
  if DTYPE != jnp.float32: