diff --git a/nodes.py b/nodes.py index 65c00a8..96c7471 100644 --- a/nodes.py +++ b/nodes.py @@ -21,6 +21,8 @@ class SaveLatentAbsolute: def save(self, samples, path): path = os.path.expanduser(path) + if not path.endswith(".latent"): + path += ".latent" os.makedirs(os.path.dirname(path), exist_ok=True) tensors = {}