steps shouldn't exceed the model's seq_len either
This commit is contained in:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user