diff --git a/src/novelai_python/sdk/ai/generate_image/__init__.py b/src/novelai_python/sdk/ai/generate_image/__init__.py index 7b1674b..1fe7b76 100755 --- a/src/novelai_python/sdk/ai/generate_image/__init__.py +++ b/src/novelai_python/sdk/ai/generate_image/__init__.py @@ -336,13 +336,18 @@ def model_post_init(self, *args) -> None: :param args: Any :return: None """ + # Must be a string if self.parameters.negative_prompt is None: self.parameters.negative_prompt = "" + + # Add negative prompt based on ucPreset if self.parameters.ucPreset is not None: default_negative_prompt = get_default_uc_preset(self.model, self.parameters.ucPreset) self.parameters.negative_prompt = ", ".join( filter(None, [default_negative_prompt, self.parameters.negative_prompt]) ) + + # Add quality prompt if self.parameters.qualityToggle: self.input += ", best quality, amazing quality, very aesthetic, absurdres"