From 9bc72acab041f471b655a763206a24513e8313d9 Mon Sep 17 00:00:00 2001 From: Ali Nehzat Date: Thu, 24 Aug 2023 09:09:16 +1000 Subject: [PATCH] steps shouldn't exceed the model's seq_len either --- run.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run.c b/run.c index da1fbc4..0eaa655 100644 --- a/run.c +++ b/run.c @@ -839,7 +839,7 @@ int main(int argc, char *argv[]) { // build the Transformer via the model .bin file Transformer transformer; build_transformer(&transformer, checkpoint_path); - if (steps == 0) steps = transformer.config.seq_len; // ovrerride to ~max length + if (steps == 0 || steps > transformer.config.seq_len) steps = transformer.config.seq_len; // ovrerride to ~max length // build the Tokenizer via the tokenizer .bin file Tokenizer tokenizer;