TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move sampling type in Model Info.

Browse Source
This commit is contained in:
Christian Bastian
2024-03-25 19:00:12 -04:00
parent a7a0a6b343
commit 764321868c
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
__init__.py
View File

@@ -632,12 +632,11 @@ async def get_model_info(request):
info["Architecture"] = metadata.get("modelspec.architecture", "") info["Architecture"] = metadata.get("modelspec.architecture", "")
info["Network Dimension"] = metadata.get("ss_network_dim", "") # features trained info["Network Dimension"] = metadata.get("ss_network_dim", "") # features trained
info["Network Alpha"] = metadata.get("ss_network_alpha", "") # trained features applied info["Network Alpha"] = metadata.get("ss_network_alpha", "") # trained features applied
info["Model Sampling Type"] = metadata.get("modelspec.prediction_type", "")
clip_skip = metadata.get("ss_clip_skip", "") clip_skip = metadata.get("ss_clip_skip", "")
if clip_skip == "None" or clip_skip == "1": # assume 1 means no clip skip if clip_skip == "None" or clip_skip == "1": # assume 1 means no clip skip
clip_skip = "" clip_skip = ""
info["Clip Skip"] = clip_skip info["Clip Skip"] = clip_skip
info["Model Sampling Type"] = metadata.get("modelspec.prediction_type", "")
# it is unclear what these are # it is unclear what these are
#info["Hash SHA256"] = metadata.get("modelspec.hash_sha256", "") #info["Hash SHA256"] = metadata.get("modelspec.hash_sha256", "")