carloscar's picture
Modifies sys.argv depending on 'IS_SHARED_UI' env value + GPU presence
42192c2 verified
raw
history blame
5.61 kB
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.&nbsp;&nbsp;<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.&nbsp;&nbsp;<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>
""")