diff --git a/modules/shared.py b/modules/shared.py index c9d85a19d..ce7f8f1bb 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -687,8 +687,8 @@ options_templates.update(options_section(('interrogate', "Interrogate"), { "interrogate_vlm_system": OptionInfo(vlm_system, "VLM: default prompt"), "interrogate_vlm_num_beams": OptionInfo(1, "VLM: num beams", gr.Slider, {"minimum": 1, "maximum": 16, "step": 1, "visible": False}), "interrogate_vlm_max_length": OptionInfo(512, "VLM: max length", gr.Slider, {"minimum": 1, "maximum": 4096, "step": 1, "visible": False}), - "interrogate_vlm_do_sample": OptionInfo(False, "VLM: use sample method"), - "interrogate_vlm_temperature": OptionInfo(0, "VLM: temperature", gr.Slider, {"minimum": 0, "maximum": 1.0, "step": 0.01, "visible": False}), + "interrogate_vlm_do_sample": OptionInfo(True, "VLM: use sample method"), + "interrogate_vlm_temperature": OptionInfo(0.8, "VLM: temperature", gr.Slider, {"minimum": 0, "maximum": 1.0, "step": 0.01, "visible": False}), "interrogate_vlm_top_k": OptionInfo(0, "VLM: top-k", gr.Slider, {"minimum": 0, "maximum": 99, "step": 1, "visible": False}), "interrogate_vlm_top_p": OptionInfo(0, "VLM: top-p", gr.Slider, {"minimum": 0, "maximum": 1.0, "step": 0.01, "visible": False}), "interrogate_vlm_keep_prefill": OptionInfo(False, "VLM: keep prefill text in output", gr.Checkbox, {"visible": False}), diff --git a/scripts/prompt_enhance.py b/scripts/prompt_enhance.py index 89df5e126..8d726f363 100644 --- a/scripts/prompt_enhance.py +++ b/scripts/prompt_enhance.py @@ -159,9 +159,9 @@ class Options: censored = ["i cannot", "i can't", "i am sorry", "against my programming", "i am not able", "i am unable", 'i am not allowed'] max_delim_index: int = 60 - max_tokens: int = 50 + max_tokens: int = 512 do_sample: bool = True - temperature: float = 0.15 + temperature: float = 0.8 repetition_penalty: float = 1.2 thinking_mode: bool = False