Merge pull request #348 from nehzata/clip_steps
Clip steps maximum value
This commit is contained in:
@@ -839,7 +839,7 @@ int main(int argc, char *argv[]) {
|
|||||||
// build the Transformer via the model .bin file
|
// build the Transformer via the model .bin file
|
||||||
Transformer transformer;
|
Transformer transformer;
|
||||||
build_transformer(&transformer, checkpoint_path);
|
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
|
// build the Tokenizer via the tokenizer .bin file
|
||||||
Tokenizer tokenizer;
|
Tokenizer tokenizer;
|
||||||
|
|||||||
Reference in New Issue
Block a user