Minor metadata bug fix (second attempt)

This commit is contained in:
oobabooga 2023-10-26 18:57:32 -07:00
parent 2d97897a25
commit 92b2f57095

View File

@ -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']