mirror of
https://github.com/deepfakes/faceswap.git
synced 2026-10-21 09:01:52 +08:00
bugfix: Make sure stored input shapes are json serializable
This commit is contained in:
@@ -341,6 +341,8 @@ class ModelBase():
|
||||
if not any(inp for inp in inputs.keys() if inp.startswith("face")):
|
||||
raise ValueError("No input named 'face' was found. Check your input naming. "
|
||||
"Current input names: {}".format(inputs))
|
||||
# Make sure they are all ints so that it can be json serialized
|
||||
inputs = {key: tuple(int(i) for i in val) for key, val in inputs.items()}
|
||||
self.state.inputs = inputs
|
||||
logger.debug("Added input shapes: %s", self.state.inputs)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user