Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
S
Stable Diffusion Webui
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Stable Diffusion Webui
Commits
fbaf6e4f
Commit
fbaf6e4f
authored
Apr 02, 2023
by
space-nuko
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Namespace metadata fields
parent
7c016dd6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
modules/extras.py
modules/extras.py
+6
-6
No files found.
modules/extras.py
View file @
fbaf6e4f
...
@@ -242,7 +242,7 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
...
@@ -242,7 +242,7 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
shared
.
state
.
textinfo
=
"Saving"
shared
.
state
.
textinfo
=
"Saving"
print
(
f
"Saving to {output_modelname}..."
)
print
(
f
"Saving to {output_modelname}..."
)
metadata
=
{
"format"
:
"pt"
,
"
models"
:
{},
"
merge_recipe"
:
None
}
metadata
=
{
"format"
:
"pt"
,
"
sd_merge_models"
:
{},
"sd_
merge_recipe"
:
None
}
if
save_metadata
:
if
save_metadata
:
merge_recipe
=
{
merge_recipe
=
{
...
@@ -260,17 +260,17 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
...
@@ -260,17 +260,17 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
"is_inpainting"
:
result_is_inpainting_model
,
"is_inpainting"
:
result_is_inpainting_model
,
"is_instruct_pix2pix"
:
result_is_instruct_pix2pix_model
"is_instruct_pix2pix"
:
result_is_instruct_pix2pix_model
}
}
metadata
[
"merge_recipe"
]
=
json
.
dumps
(
merge_recipe
)
metadata
[
"
sd_
merge_recipe"
]
=
json
.
dumps
(
merge_recipe
)
def
add_model_metadata
(
checkpoint_info
):
def
add_model_metadata
(
checkpoint_info
):
checkpoint_info
.
calculate_shorthash
()
checkpoint_info
.
calculate_shorthash
()
metadata
[
"models"
][
checkpoint_info
.
sha256
]
=
{
metadata
[
"
sd_merge_
models"
][
checkpoint_info
.
sha256
]
=
{
"name"
:
checkpoint_info
.
name
,
"name"
:
checkpoint_info
.
name
,
"legacy_hash"
:
checkpoint_info
.
hash
,
"legacy_hash"
:
checkpoint_info
.
hash
,
"
merge_recipe"
:
checkpoint_info
.
metadata
.
get
(
"
merge_recipe"
,
None
)
"
sd_merge_recipe"
:
checkpoint_info
.
metadata
.
get
(
"sd_
merge_recipe"
,
None
)
}
}
metadata
[
"
models"
]
.
update
(
checkpoint_info
.
metadata
.
get
(
"
models"
,
{}))
metadata
[
"
sd_merge_models"
]
.
update
(
checkpoint_info
.
metadata
.
get
(
"sd_merge_
models"
,
{}))
add_model_metadata
(
primary_model_info
)
add_model_metadata
(
primary_model_info
)
if
secondary_model_info
:
if
secondary_model_info
:
...
@@ -278,7 +278,7 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
...
@@ -278,7 +278,7 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
if
tertiary_model_info
:
if
tertiary_model_info
:
add_model_metadata
(
tertiary_model_info
)
add_model_metadata
(
tertiary_model_info
)
metadata
[
"
models"
]
=
json
.
dumps
(
metadata
[
"
models"
])
metadata
[
"
sd_merge_models"
]
=
json
.
dumps
(
metadata
[
"sd_merge_
models"
])
_
,
extension
=
os
.
path
.
splitext
(
output_modelname
)
_
,
extension
=
os
.
path
.
splitext
(
output_modelname
)
if
extension
.
lower
()
==
".safetensors"
:
if
extension
.
lower
()
==
".safetensors"
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment