aeb516494daa207720b428072ab49eeea7bfff75,scripts/text_generation/sequence_sampling.py,,,#,33

Before Change

if isinstance(args.bos, str):
    args.bos = [args.bos]

lm_model, vocab = nlp.model.get_model(name=args.lm_model,
                                      dataset_name="wikitext-2",
                                      pretrained=True,
                                      ctx=ctx)

After Change

                                             help="Use random sampling for decoding.")
random_sample_parser.add_argument("--temperature", type=float, default=1.0,
                                  help="Softmax temperature used in sampling.")
random_sample_parser.add_argument("--use-top-k", type=int, required=False,
                                  help="Sample only from the top-k candidates.")

// shared options
for p in [beam_search_parser, random_sample_parser]:
    p.add_argument("--gpu", type=int, default=0,

Instances