bofenghuang commited on
Commit
025dcb5
·
1 Parent(s): 1c16508
Files changed (1) hide show
  1. run_demo_multi_models.py +1 -1
run_demo_multi_models.py CHANGED
@@ -60,7 +60,7 @@ def maybe_load_cached_pipeline(model_name):
60
  # set forced_decoder_ids
61
  pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language="fr", task="transcribe")
62
  # limit genneration max length
63
- pipe.model.config.max_length = MAX_NEW_TOKENS + 1
64
 
65
  logger.info(f"`{model_name}` pipeline has been initialized")
66
  print_memory_info()
 
60
  # set forced_decoder_ids
61
  pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language="fr", task="transcribe")
62
  # limit genneration max length
63
+ # pipe.model.config.max_length = MAX_NEW_TOKENS + 1
64
 
65
  logger.info(f"`{model_name}` pipeline has been initialized")
66
  print_memory_info()