Update _type_mapping, allow .copy() to work.

Aedial · Dec 3, 2023 · 52df5f2 · 52df5f2
1 parent 03d3a5a
commit 52df5f2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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__