ChenWu98 commited on
Commit
7138b0a
β€’
1 Parent(s): 820a643

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -9,7 +9,9 @@ is_colab = utils.is_google_colab()
9
 
10
  if True:
11
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
12
- scheduler = DDIMScheduler.from_config(model_id_or_path, subfolder="scheduler")
 
 
13
  pipe = CycleDiffusionPipeline.from_pretrained(model_id_or_path,
14
  use_auth_token=st.secrets["USER_TOKEN"],
15
  scheduler=scheduler)
@@ -59,7 +61,7 @@ with gr.Blocks(css=css) as demo:
59
  </div>
60
  <p>
61
  Demo for CycleDiffusion with Stable Diffusion, built with Diffusers 🧨 by HuggingFace πŸ€—. <br>
62
- <a href="https://huggingface.co/docs/diffusers/main/en/api/pipelines/cycle_diffusion">Doc</a> | <a href="https://arxiv.org/abs/2210.05559">Paper</a>
63
  </p>
64
  <p>You can skip the queue in the colab: <a href="https://colab.research.google.com/gist/ChenWu98/0aa4fe7be80f6b45d3d055df9f14353a/copy-of-fine-tuned-diffusion-gradio.ipynb"><img data-canonical-src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" src="https://colab.research.google.com/assets/colab-badge.svg"></a></p>
65
  Running on <b>{device}</b>{(" in a <b>Google Colab</b>." if is_colab else "")}
 
9
 
10
  if True:
11
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
12
+ scheduler = DDIMScheduler.from_config(model_id_or_path,
13
+ use_auth_token=st.secrets["USER_TOKEN"],
14
+ subfolder="scheduler")
15
  pipe = CycleDiffusionPipeline.from_pretrained(model_id_or_path,
16
  use_auth_token=st.secrets["USER_TOKEN"],
17
  scheduler=scheduler)
 
61
  </div>
62
  <p>
63
  Demo for CycleDiffusion with Stable Diffusion, built with Diffusers 🧨 by HuggingFace πŸ€—. <br>
64
+ <a href="https://huggingface.co/docs/diffusers/main/en/api/pipelines/cycle_diffusion">Pipeline doc</a> | <a href="https://arxiv.org/abs/2210.05559">Paper</a>
65
  </p>
66
  <p>You can skip the queue in the colab: <a href="https://colab.research.google.com/gist/ChenWu98/0aa4fe7be80f6b45d3d055df9f14353a/copy-of-fine-tuned-diffusion-gradio.ipynb"><img data-canonical-src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" src="https://colab.research.google.com/assets/colab-badge.svg"></a></p>
67
  Running on <b>{device}</b>{(" in a <b>Google Colab</b>." if is_colab else "")}