ChenWu98 commited on
Commit
fd8f4d5
1 Parent(s): 3728d9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -13,7 +13,8 @@ LOW_RESOURCE = False
13
  MAX_NUM_WORDS = 77
14
 
15
  is_colab = utils.is_google_colab()
16
-
 
17
 
18
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
19
  if is_colab:
@@ -311,7 +312,9 @@ with gr.Blocks(css=css) as demo:
311
  3. If CAC type is "Replace", the source and target prompts should differ in only one token; otherwise, an error will be raised. This is why we deliberately make some grammar mistakes in Examples.<br>
312
  4. If CAC type is "Refine", the source prompt be a subsequence of the target prompt; otherwise, an error will be raised. <br>
313
  </p>
314
- <p>You can skip the queue using 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>
 
 
315
  Running on <b>{device_print}</b>{(" in a <b>Google Colab</b>." if is_colab else "")}
316
  </p>
317
  </div>
 
13
  MAX_NUM_WORDS = 77
14
 
15
  is_colab = utils.is_google_colab()
16
+ colab_instruction = """
17
+ <p>You can skip the queue using 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>""" if is_colab else ""
18
 
19
  model_id_or_path = "CompVis/stable-diffusion-v1-4"
20
  if is_colab:
 
312
  3. If CAC type is "Replace", the source and target prompts should differ in only one token; otherwise, an error will be raised. This is why we deliberately make some grammar mistakes in Examples.<br>
313
  4. If CAC type is "Refine", the source prompt be a subsequence of the target prompt; otherwise, an error will be raised. <br>
314
  </p>
315
+ <p>
316
+ <b>Runtimes:</b> 30s on A10G small, 80s on T4 small. <br>
317
+ {colab_instruction}
318
  Running on <b>{device_print}</b>{(" in a <b>Google Colab</b>." if is_colab else "")}
319
  </p>
320
  </div>