Minor metadata bug fix

This commit is contained in:
oobabooga 2023-10-23 12:46:24 -07:00
parent 92691ee626
commit 306d764ff6

View file

@ -91,9 +91,9 @@ def create_event_handlers():
def get_truncation_length():
if shared.args.max_seq_len != shared.args_defaults.max_seq_len:
if 'max_seq_len' in shared.provided_arguments:
return shared.args.max_seq_len
if shared.args.n_ctx != shared.args_defaults.n_ctx:
elif 'n_ctx' in shared.provided_arguments:
return shared.args.n_ctx
else:
return shared.settings['truncation_length']