Skip to content

Commit

Permalink
[API] Fix name for Scribbler and Landscaper
Browse files Browse the repository at this point in the history
  • Loading branch information
Aedial committed Jul 10, 2023
1 parent ad281b5 commit e6286e1
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions novelai_api/ImagePreset.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,9 +30,9 @@ class ControlNetModel(enum.Enum):

Palette_Swap = "hed"
Form_Lock = "midas"
Scrible = "fake_scribble"
Scribbler = "fake_scribble"
Building_Control = "mlsd"
Lanscaper = "uniformer"
Landscaper = "uniformer"


class ImageResolution(enum.Enum):
Expand Down Expand Up @@ -136,9 +136,9 @@ class ImagePreset:
_CONTROLNET_MODELS = {
ControlNetModel.Palette_Swap: "hed",
ControlNetModel.Form_Lock: "depth",
ControlNetModel.Scrible: "scribble",
ControlNetModel.Scribbler: "scribble",
ControlNetModel.Building_Control: "mlsd",
ControlNetModel.Lanscaper: "seg",
ControlNetModel.Landscaper: "seg",
}

_TYPE_MAPPING = {
Expand Down

0 comments on commit e6286e1

Please sign in to comment.