Hugging Face's logo

Spaces:
bguisard
/
stable-diffusion-nano
Runtime error

App Files Community
2
bguisard commited on
Commit
aa9e215
·
1 Parent(s): 9e5e7f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from diffusers import FlaxPNDMScheduler, FlaxStableDiffusionPipeline
5
  from flax.jax_utils import replicate
6
  from flax.training.common_utils import shard
7
 
8
- DTYPE = jnp.float32
9
 
10
  pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
11
  "bguisard/stable-diffusion-nano-2-1",
 
5
  from flax.jax_utils import replicate
6
  from flax.training.common_utils import shard
7
 
8
+ DTYPE = jnp.float16
9
 
10
  pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
11
  "bguisard/stable-diffusion-nano-2-1",