stable-diffusion-textual-in.../generate_tx_models_html.py

243 lines
10 KiB
Python
Raw Permalink Normal View History

# pip install html5print requests huggingface-hub Pillow
2022-09-24 21:25:08 -06:00
import argparse
import datetime
import os
import shutil
import sys
from urllib import request as ulreq
import requests
from huggingface_hub import HfApi
from PIL import ImageFile
parser = argparse.ArgumentParser()
parser.add_argument('out_file', nargs='?', help='file to save to', default='stable-diffusion-textual-inversion-models.html')
args = parser.parse_args()
print('Will save to file:', args.out_file)
2022-09-24 22:52:28 -06:00
# Init some stuff before saving the time
2022-09-24 21:25:08 -06:00
api = HfApi()
models_list = []
2022-09-24 22:52:28 -06:00
# Save the time now before we do the hard work
2022-09-24 21:25:08 -06:00
dt = datetime.datetime.now()
tz = dt.astimezone().tzname()
2022-09-24 22:52:28 -06:00
# Get list of models under the sd-concepts-library organization
for model in api.list_models(author="sd-concepts-library"):
models_list.append(model.modelId.replace('sd-concepts-library/', ''))
models_list.sort()
2022-09-24 21:25:08 -06:00
html_struct = f"""
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8">
2022-09-30 10:06:54 -06:00
<title>Stable Diffusion Textual Inversion Embeddings</title>
2022-09-24 21:25:08 -06:00
<meta http-equiv="Cache-Control" content="no-cache, no-store, must-revalidate" />
<meta http-equiv="Pragma" content="no-cache" />
<meta http-equiv="Expires" content="0" />
2022-09-25 23:52:10 -06:00
<meta name="viewport" content="width=device-width, initial-scale=1" />
2022-09-30 22:51:22 -06:00
<meta name="google-site-verification" content="pgoBH8R__ZWngQ-S3o8xopSpHYROu_0VfoS5VYXB3uw" />
2022-10-01 10:50:23 -06:00
2022-09-24 21:25:08 -06:00
<script src="https://cdn.jsdelivr.net/npm/axios/dist/axios.min.js"></script>
<link href="https://cdn.jsdelivr.net/npm/bootstrap@5.2.1/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-iYQeCzEYFbKjA/T2uDLTpkwGzCiq6soy8tYaI1GyVh/UjpbCx/TYkiZhlZB6+fzT" crossorigin="anonymous">
<link rel="apple-touch-icon" sizes="180x180" href="/stable-diffusion-textual-inversion-models/apple-touch-icon.png">
<link rel="icon" type="image/png" sizes="32x32" href="/stable-diffusion-textual-inversion-models/favicon-32x32.png">
<link rel="icon" type="image/png" sizes="16x16" href="/stable-diffusion-textual-inversion-models/favicon-16x16.png">
<link rel="manifest" href="/stable-diffusion-textual-inversion-models/site.webmanifest">
<link rel="mask-icon" href="/stable-diffusion-textual-inversion-models/safari-pinned-tab.svg" color="#ee9321">
<link rel="shortcut icon" href="favicon.ico">
<meta name="msapplication-TileColor" content="#ee9321">
<meta name="msapplication-config" content="/stable-diffusion-textual-inversion-models/browserconfig.xml">
<meta name="theme-color" content="#ee9321">
<!-- Matomo -->
2022-09-25 23:52:10 -06:00
<!-- Open-source, self hosted, private analytics. No third parties involved. -->
2022-09-24 21:25:08 -06:00
<script>
var _paq = window._paq = window._paq || [];
_paq.push(['trackPageView']);
2022-09-25 19:17:37 -06:00
_paq.push(['trackVisibleContentImpressions']);
2022-09-24 21:25:08 -06:00
_paq.push(['enableLinkTracking']);
2022-09-25 23:52:10 -06:00
_paq.push(['enableHeartBeatTimer']);
2022-09-24 21:25:08 -06:00
(function() {{
var u = "https://mato.evulid.cc/";
_paq.push(['setTrackerUrl', u + 'matomo.php']);
_paq.push(['setSiteId', '1']);
var d = document,
g = d.createElement('script'),
s = d.getElementsByTagName('script')[0];
g.async = true;
g.src = u + 'matomo.js';
s.parentNode.insertBefore(g, s);
}})();
</script>
<!-- End Matomo Code -->
</head>
<body>
<style>
.thumbnail {{
max-width: 185px;
display: block;
padding-top: 5px;
padding-bottom: 5px;
}}
.model-title {{
margin-top: 100px;
}}
2022-10-07 16:34:25 -06:00
body {{
2022-10-07 16:39:11 -06:00
background-color: #0000ff0d !important;
2022-10-07 16:34:25 -06:00
}}
2022-10-07 17:46:39 -06:00
.model-title > a {{
color: initial !important;
text-decoration: none !important;
}}
2022-09-24 21:25:08 -06:00
</style>
2022-09-25 23:59:31 -06:00
2022-10-21 13:59:31 -06:00
<nav class="navbar navbar-expand-lg navbar-light bg-light" style="background-color:#6c757d42!important">
<div class="container-fluid"><button class="navbar-toggler" type="button" data-bs-toggle="collapse" data-bs-target="#navtogglecontent" aria-controls="navbarToggleExternalContent" aria-expanded="false" aria-label="Toggle navigation"><span
class="navbar-toggler-icon"></span></button>
<div class="collapse navbar-collapse" id="navtogglecontent">
2022-09-25 23:52:10 -06:00
<div class="navbar-nav">
<a class="nav-link" href="/stable-diffusion-models/">Models</a>
2022-10-21 13:59:31 -06:00
<a class="nav-link active" aria-current="page" href="/stable-diffusion-textual-inversion-models/">Textual Inversion Models</a>
<a class="nav-link" href="/stable-diffusion-dreambooth-library/">DreamBooth Models</a>
<a class="nav-link" href="/stable-diffusion-models/sdmodels/">sdmodels</a>
2022-09-25 23:52:10 -06:00
</div>
</div>
</div>
</nav>
2022-09-25 23:59:31 -06:00
2022-09-24 21:25:08 -06:00
<div class="container" style="margin-bottom: 180px;">
<div class="jumbotron text-center" style="margin-top: 45px;margin-right: 45px;margin-bottom: 0px;margin-left: 45px;">
2022-09-30 10:06:54 -06:00
<h1>Stable Diffusion Textual Inversion Embeddings</h1>
2022-09-24 21:25:08 -06:00
</div>
2022-09-24 22:52:28 -06:00
<div style="text-align: center;margin-bottom: 45px;font-size: 8pt;">
2022-09-24 22:57:00 -06:00
<p>
2022-10-07 16:39:11 -06:00
<i>Page updates automatically daily. Last updated <a class="btn-link" style="cursor: pointer;text-decoration: none;" data-toggle="tooltip" data-placement="bottom" title="{dt.strftime(f"%m-%d-%Y %H:%M:%S {tz}")}">{dt.strftime("%A %B %d, %Y")}</a>.</i>
2022-09-24 21:25:08 -06:00
</p>
2022-09-24 22:52:28 -06:00
</div>
2022-09-24 21:25:08 -06:00
<p>
2022-10-07 16:49:20 -06:00
Browser for the <a href="https://huggingface.co/sd-concepts-library">HuggingFace textual inversion library</a>. There are currently {len(models_list)} textual inversion embeddings in sd-concepts-library. These are meant to be used with <a href="https://github.com/AUTOMATIC1111/stable-diffusion-webui">AUTOMATIC1111's SD WebUI</a>.
2022-09-24 21:25:08 -06:00
</p>
2022-09-25 23:59:31 -06:00
2022-09-24 21:25:08 -06:00
<p>
Embeddings are downloaded straight from the HuggingFace repositories. The images displayed are the inputs, not the outputs. Want to quickly test concepts? Try the <a href="https://huggingface.co/spaces/sd-concepts-library/stable-diffusion-conceptualizer">Stable Diffusion Conceptualizer</a> on HuggingFace. <a href="https://huggingface.co/docs/diffusers/main/en/training/text_inversion">More info on textual inversion.</a>
2022-09-25 23:59:31 -06:00
</p>
2022-09-24 23:03:17 -06:00
<center>
2022-09-24 21:25:08 -06:00
<a href="https://github.com/Cyberes/stable-diffusion-textual-inversion-models/actions/workflows/generate_static_html.yml"><img src="https://github.com/Cyberes/stable-diffusion-textual-inversion-models/actions/workflows/generate_static_html.yml/badge.svg"></a>
2022-09-24 23:03:17 -06:00
</center>
2022-09-24 21:25:08 -06:00
<br>
<hr>
<noscript><p><img src="https://mato.evulid.cc/matomo.php?idsite=1&rec=1&url=https://cyberes.github.io/stable-diffusion-textual-inversion-models" style="border:0;" alt="" /></p></noscript>
"""
2022-09-25 23:59:31 -06:00
2022-09-24 21:25:08 -06:00
i = 1
for model_name in models_list:
# For testing
# if i == 4:
# break
2022-09-24 21:25:08 -06:00
print(f'{i}/{len(models_list)} -> {model_name}')
2022-09-25 23:59:31 -06:00
2022-10-07 17:59:04 -06:00
html_struct = html_struct + f'<div data-track-content data-content-name="{model_name}" data-content-piece="TX Model Item"><h3 class="model-title" id="{model_name}"><a href="#{model_name}">{model_name}</a></h3>'
2022-09-24 21:25:08 -06:00
2022-09-24 22:52:28 -06:00
# Get the concept images from the huggingface repo
2022-09-24 21:25:08 -06:00
restricted = False
try:
files = api.list_repo_files(
repo_id=f'sd-concepts-library/{model_name}')
concept_images = [i for i in files if i.startswith('concept_images/')]
except requests.exceptions.HTTPError:
2022-09-24 22:52:28 -06:00
# Sometimes an author will require you to share your contact info to gain access
2022-09-24 21:25:08 -06:00
restricted = True
if restricted:
html_struct = html_struct + f"""
<p>
{model_name} is restricted and you must share your contact information to view this repository.
<a type="button" class="btn btn-link" href="https://huggingface.co/sd-concepts-library/{model_name}/">View Repository</a>
</p>
"""
else:
html_struct = html_struct + f"""
<p>
<button type="button" class="btn btn-primary" onclick="downloadAs('https://huggingface.co/sd-concepts-library/{model_name}/resolve/main/learned_embeds.bin', '{model_name}.pt')">Download {model_name}.pt</button>
<a type="button" class="btn btn-link" href="https://huggingface.co/sd-concepts-library/{model_name}/">View Repository</a>
</p>
<div class="row">
"""
2022-09-24 22:52:28 -06:00
# Most repos have 3 concept images but some have more or less
# We gotta make sure only 3 are shown
2022-09-24 21:25:08 -06:00
img_count = 3
if len(concept_images) < 3:
img_count = len(concept_images)
for x in range(img_count):
html_struct = html_struct + f"""
<div class="col-sm">
2022-09-24 22:52:28 -06:00
<!-- <img class="thumbnail mx-auto lazy-load img-fluid" data-src="https://huggingface.co/sd-concepts-library/{model_name}/resolve/main/{concept_images[x]}">-->
<img class="thumbnail mx-auto img-fluid" loading="lazy" src="https://huggingface.co/sd-concepts-library/{model_name}/resolve/main/{concept_images[x]}">
2022-09-24 21:25:08 -06:00
</div>
"""
2022-09-26 12:50:18 -06:00
html_struct = html_struct + '</div></div>'
2022-09-24 21:25:08 -06:00
i = i + 1
html_struct = html_struct + """
</div>
<script>
2022-09-24 22:52:28 -06:00
// Download the file under a different name
const downloadAs = (url, name) => {
axios.get(url, {
headers: {
"Content-Type": "application/octet-stream"
},
responseType: "blob"
})
.then(response => {
const a = document.createElement("a");
const url = window.URL.createObjectURL(response.data);
a.href = url;
a.download = name;
a.click();
})
.catch(err => {
console.log("error", err);
});
_paq.push(['trackLink', url, 'download']);
};
2022-09-26 00:04:47 -06:00
document.addEventListener("DOMContentLoaded", () => {
2022-09-26 00:12:26 -06:00
// Enable tooltips
$(function() {
$('[data-toggle="tooltip"]').tooltip({
placement: "bottom"
})
});
2022-09-26 00:27:52 -06:00
/*
2022-09-25 23:59:31 -06:00
var waypoints = $(".model-title").waypoint(function(direction) {
2022-09-26 00:14:41 -06:00
_paq.push(["trackEvent", "Scroll", "View TX Model", this.element.getAttribute("data-content-name")]);
2022-09-25 23:59:31 -06:00
})
2022-09-26 00:27:52 -06:00
*/
2022-09-26 00:04:47 -06:00
});
2022-09-24 21:25:08 -06:00
</script>
2022-09-25 23:59:31 -06:00
<script src="https://cdn.jsdelivr.net/npm/bootstrap@5.2.1/dist/js/bootstrap.bundle.min.js" integrity="sha384-u1OknCvxWvY5kfmNBILK2hRnQC3Pr17a+RTT6rIHI7NnikvbZlHgTPOOmMi466C8" crossorigin="anonymous"></script>
<script src="https://cdnjs.cloudflare.com/ajax/libs/jquery/3.5.1/jquery.min.js" integrity="sha512-bLT0Qm9VnAYZDflyKcBaQ2gg0hSYNQrJ8RilYldYQ1FxQYoCLtUjuuRuZo+fjqhx/qtq/1itJ0C2ejDxltZVFg==" crossorigin="anonymous"></script>
2022-10-07 16:49:20 -06:00
<!--<script src="/stable-diffusion-textual-inversion-models/jquery.waypoints.min.js"></script>-->
2022-10-21 14:02:50 -06:00
<script src="https://getbootstrap.com/docs/5.0/dist/js/bootstrap.bundle.min.js"></script>
2022-09-24 21:25:08 -06:00
</body>
"""
f = open(args.out_file, 'w', encoding='utf-8')
f.write(html_struct)
2022-09-24 21:25:08 -06:00
f.close()