Spaces:
Sleeping
Sleeping
bofenghuang
commited on
Commit
·
025dcb5
1
Parent(s):
1c16508
up
Browse files- 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()
|