add a bit more metadata info for the lora user metadata page

This commit is contained in:
AUTOMATIC1111 2023-07-18 17:32:46 +03:00
parent ed82f1c5f1
commit 4b5a63aa11
1 changed files with 6 additions and 0 deletions

View File

@ -1,3 +1,4 @@
import datetime
import html import html
import random import random
@ -71,6 +72,7 @@ class LoraUserMetadataEditor(ui_extra_networks_user_metadata.UserMetadataEditor)
keys = { keys = {
'ss_sd_model_name': "Model:", 'ss_sd_model_name': "Model:",
'ss_clip_skip': "Clip skip:", 'ss_clip_skip': "Clip skip:",
'ss_network_module': "Kohya module:",
} }
for key, label in keys.items(): for key, label in keys.items():
@ -78,6 +80,10 @@ class LoraUserMetadataEditor(ui_extra_networks_user_metadata.UserMetadataEditor)
if value is not None and str(value) != "None": if value is not None and str(value) != "None":
table.append((label, html.escape(value))) table.append((label, html.escape(value)))
ss_training_started_at = metadata.get('ss_training_started_at')
if ss_training_started_at:
table.append(("Date trained:", datetime.datetime.utcfromtimestamp(float(ss_training_started_at)).strftime('%Y-%m-%d %H:%M')))
ss_bucket_info = metadata.get("ss_bucket_info") ss_bucket_info = metadata.get("ss_bucket_info")
if ss_bucket_info and "buckets" in ss_bucket_info: if ss_bucket_info and "buckets" in ss_bucket_info:
resolutions = {} resolutions = {}