'TextGenerationPipeline' has no attribute ALLOWED_MODELS

[mrseeker]

This part looks broken on Huggingface Transformers main build:

mkultra/mkultra/inference.py

Line 12 in a25c72d

if model not in TextGenerationPipeline.ALLOWED_MODELS:

[gungunyuanda]

Hi, I met the same problem, have you solved it? Thanks.

[ebolam]

It appears to work if you comment out that section (lines 11, 12, and 13). Might cause some error checking to fail later though depending on your use case, but I was able to run the text_generation.ipynb example.

[Tomik292]

@ebolam answer did not work for me. The code fails on

generator = pipeline('text-generation', model=model, tokenizer=tokenizer)

with

RuntimeError: Failed to import transformers.models.biogpt.configuration_biogpt because of the following error (look up to see its traceback):
No module named 'transformers.models.biogpt.configuration_biogpt'

[Tomik292]

I was able to the fix the problem using older version of transformers==4.15.0.