apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

prep models field

This commit is contained in:
Sean Sube 2024-01-03 23:38:44 -06:00
parent 04be627e7b
commit fe6b267483
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/onnx_web/chain/result.py
View File

@ -127,8 +127,15 @@ class ImageMetadata:
} }
# fix up some fields # fix up some fields
json["params"]["model"] = path.basename(self.params.model) model = path.basename(self.params.model)
json["params"]["scheduler"] = self.params.scheduler # TODO: why tho? json["params"]["model"] = model
json["models"].append(
{
"name": model,
"weight": 1.0,
"hash": self.get_model_hash(model),
}
)
# calculate final output size # calculate final output size
output_size = self.size output_size = self.size