Text Generation in Colab

[elliotthwang]

Hi !
Text Generation in Colab
AttributeError Traceback (most recent call last)
in ()
1 from transformers.pipelines import pipeline
----> 2 from mkultra.inference import GPT2SoftPromptLM
3 from mkultra.tokenizers import GPT2SPTokenizerFast
4 from mkultra.soft_prompt import SoftPrompt
5 import torch

/usr/local/lib/python3.7/dist-packages/mkultra/inference.py in ()
10
11 for model in EXTRA_ALLOWED_MODELS:
---> 12 if model not in TextGenerationPipeline.ALLOWED_MODELS:
13 TextGenerationPipeline.ALLOWED_MODELS.append(model)
14

AttributeError: type object 'TextGenerationPipeline' has no attribute 'ALLOWED_MODELS'

[Tomik292]

I am facing the same issue!

[Tomik292]

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