diff --git a/modules/ui_parameters.py b/modules/ui_parameters.py index cd819a62..808b81f2 100644 --- a/modules/ui_parameters.py +++ b/modules/ui_parameters.py @@ -92,9 +92,9 @@ def create_event_handlers(): def get_truncation_length(): - if 'max_seq_len' in shared.provided_arguments: + if 'max_seq_len' in shared.provided_arguments or shared.args.max_seq_len != shared.args_defaults.max_seq_len: return shared.args.max_seq_len - elif 'n_ctx' in shared.provided_arguments: + elif 'n_ctx' in shared.provided_arguments or shared.args.n_ctx != shared.args_defaults.n_ctx: return shared.args.n_ctx else: return shared.settings['truncation_length']