From 52df5f26b115c0e37acf438bd62341edf408916e Mon Sep 17 00:00:00 2001 From: "agentd00nut@gmail.com" Date: Sat, 2 Dec 2023 21:01:37 -0500 Subject: [PATCH] Update _type_mapping, allow .copy() to work. --- novelai_api/ImagePreset.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/novelai_api/ImagePreset.py b/novelai_api/ImagePreset.py index e737757..bdacc8b 100644 --- a/novelai_api/ImagePreset.py +++ b/novelai_api/ImagePreset.py @@ -199,6 +199,7 @@ class ImagePreset: } _TYPE_MAPPING = { + "legacy": bool, "quality_toggle": bool, "resolution": (ImageResolution, tuple), "uc_preset": (UCPreset, NoneType), @@ -208,6 +209,7 @@ class ImagePreset: "noise": (int, float), "strength": (int, float), "scale": (int, float), + "uncond_scale": (int, float), "steps": int, "uc": str, "smea": bool, @@ -219,6 +221,8 @@ class ImagePreset: "decrisper": bool, "add_original_image": bool, "mask": str, + "cfg_rescale": float, + "noise_schedule": str, } # type completion for __setitem__ and __getitem__