transformers-community

/

constrained-beam-search

like 1

Follow

Transformers Community 67

Add a default value to the argument synced_gpus to avoid this error:

Traceback (most recent call last):
  File "myFolder/src/translationProject/main.py", line 12, in <module>
    outputs = model.generate(
  File "myFolder/.venv/lib64/python3.9/site-packages/torch/utils/_contextlib.py", line 120, in decorate_context
    return func(*args, **kwargs)
  File "myFolder/.venv/lib/python3.9/site-packages/transformers/generation/utils.py", line 2367, in generate
    return custom_generate_function(model=self, **generate_arguments)
  File "/root/.cache/huggingface/modules/transformers_modules/transformers_hyphen_community/constrained_hyphen_beam_hyphen_search/3081418faf290f61bc253e649b0033adf877e655/custom_generate/generate.py", line 338, in generate
    generation_outputs = GenerationMixin.generate(
  File "myFolder/.venv/lib64/python3.9/site-packages/torch/utils/_contextlib.py", line 120, in decorate_context
    return func(*args, **kwargs)
  File "myFolder/.venv/lib/python3.9/site-packages/transformers/generation/utils.py", line 2564, in generate
    result = decoding_method(
TypeError: _constrained_beam_search() missing 1 required positional argument: 'synced_gpus'