ash123 commited on
Commit
1f33c8b
1 Parent(s): 6ceebc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import jax
3
  import jax.numpy as jnp
4
- from diffusers import FlaxStableDiffusionPipeline
5
  from flax.jax_utils import replicate
6
  from flax.training.common_utils import shard
7
  from share_btn import community_icon_html, loading_icon_html, share_js
 
1
  import gradio as gr
2
  import jax
3
  import jax.numpy as jnp
4
+ from diffusers import FlaxPNDMScheduler, FlaxStableDiffusionPipeline
5
  from flax.jax_utils import replicate
6
  from flax.training.common_utils import shard
7
  from share_btn import community_icon_html, loading_icon_html, share_js