-
Notifications
You must be signed in to change notification settings - Fork 6
/
header_patch.py
39 lines (39 loc) Β· 5.48 KB
/
header_patch.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
with gr.Box(visible=os.environ.get("SPACE_ID")):
is_shared_ui = str(os.environ.get("IS_SHARED_UI", "") or "").strip().lower() not in ("", "0", "false", "none", "no")
if is_shared_ui:
import torch
if not torch.cuda.is_available():
gr.HTML(f"""
<div class="gr-prose" style="max-width: 80%; font-size: 12px; line-height: 20px; font-family: monospace;">
<p>β² Automatic1111's Stable Diffusion WebUI + Mikubill's ControlNet WebUI extension | Running on Hugging Face | Loaded checkpoint: <a href="https://civitai.com/models/8124/a-to-zovya-rpg-artists-tools-15-and-21" style="target=" _blank"="">AtoZovyaRPGArtistTools15_sd15V1</a></p>
<p>β² Docker build from <a href="https://github.com/kalaspuff/stable-diffusion-webui-controlnet-docker" style="target=" _blank"="">π GitHub β kalaspuff/stable-diffusion-webui-controlnet-docker</a> / <a href="https://huggingface.co/spaces/carloscar/stable-diffusion-webui-controlnet-docker" style="target=" _blank"="">π€ Hugging Face β carloscar/stable-diffusion-webui-controlnet-docker</a></p>
<p>β² This Space is currently running on CPU, which may yield very slow results - you can upgrade for a GPU after duplicating the space.</p>
<p>β² Duplicate this Space to run it privately without a queue, use a GPU for faster generation times, load custom checkpoints, etc. <a style="display:inline-block; position: absolute;" target="_blank" href="https://huggingface.co/spaces/{os.environ["SPACE_ID"]}?duplicate=true"><img style="margin: 0; height: 16px; position: relative; top: 2px;" src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Duplicate%20Space-blue" alt="Duplicate Space"></a></p>
</div>
""")
else:
gr.HTML(f"""
<div class="gr-prose" style="max-width: 80%; font-size: 12px; line-height: 20px; font-family: monospace;">
<p>β² Automatic1111's Stable Diffusion WebUI + Mikubill's ControlNet WebUI extension | Running on Hugging Face | Loaded checkpoint: <a href="https://civitai.com/models/8124/a-to-zovya-rpg-artists-tools-15-and-21" style="target=" _blank"="">AtoZovyaRPGArtistTools15_sd15V1</a></p>
<p>β² Docker build from <a href="https://github.com/kalaspuff/stable-diffusion-webui-controlnet-docker" style="target=" _blank"="">π GitHub β kalaspuff/stable-diffusion-webui-controlnet-docker</a> / <a href="https://huggingface.co/spaces/carloscar/stable-diffusion-webui-controlnet-docker" style="target=" _blank"="">π€ Hugging Face β carloscar/stable-diffusion-webui-controlnet-docker</a></p>
<p>β² Duplicate this Space to run it privately without a queue, use extensions, load custom checkpoints, etc. <a style="display:inline-block; position: absolute;" target="_blank" href="https://huggingface.co/spaces/{os.environ["SPACE_ID"]}?duplicate=true"><img style="margin: 0; height: 16px; position: relative; top: 2px;" src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Duplicate%20Space-blue" alt="Duplicate Space"></a></p>
</div>
""")
elif os.environ.get("SPACE_ID"):
import torch
if not torch.cuda.is_available():
gr.HTML(f"""
<div class="gr-prose" style="max-width: 80%; font-size: 12px; line-height: 20px; font-family: monospace;">
<p>β² Docker build from <a href="https://github.com/kalaspuff/stable-diffusion-webui-controlnet-docker" style="target=" _blank"="">π GitHub β kalaspuff/stable-diffusion-webui-controlnet-docker</a> / <a href="https://huggingface.co/spaces/carloscar/stable-diffusion-webui-controlnet-docker" style="target=" _blank"="">π€ Hugging Face β carloscar/stable-diffusion-webui-controlnet-docker</a></p>
<p>β² Load additional checkpoints, VAE, LoRA models, etc. Read more on the README at the GitHub link above.</p>
<p>β² This Space is currently running on CPU, which may yield very slow results - you can upgrade for a GPU <a href="https://huggingface.co/spaces/{os.environ["SPACE_ID"]}/settings" style="text-decoration: underline" target="_blank">in the Settings tab</a></p>
</div>
""")
else:
gr.HTML(f"""
<div class="gr-prose" style="max-width: 80%; font-size: 12px; line-height: 20px; font-family: monospace;">
<p>β² Docker build from <a href="https://github.com/kalaspuff/stable-diffusion-webui-controlnet-docker" style="target=" _blank"="">π GitHub β kalaspuff/stable-diffusion-webui-controlnet-docker</a> / <a href="https://huggingface.co/spaces/carloscar/stable-diffusion-webui-controlnet-docker" style="target=" _blank"="">π€ Hugging Face β carloscar/stable-diffusion-webui-controlnet-docker</a></p>
<p>β² Load additional checkpoints, VAE, LoRA models, etc. Read more on the README at the GitHub link above.</p>
<p>β² This Space has GPU enabled - remember to remove the GPU from the space <a href="https://huggingface.co/spaces/{os.environ["SPACE_ID"]}/settings" style="text-decoration: underline" target="_blank">in the Settings tab</a> when you're done.</p>
</div>
""")