| 50 | |
| 51 | |
| 52 | class DummyTextGenerationPipeline: |
| 53 | def __init__(self, prefix="generated", task="text-generation", num_return_sequences=1): |
| 54 | self.task = task |
| 55 | self.prefix = prefix |
| 56 | self.num_return_sequences = num_return_sequences |
| 57 | |
| 58 | def __call__(self, inputs, **kwargs): |
| 59 | return [[{f"{self.prefix}_text": "Lorem ipsum"} for _ in range(self.num_return_sequences)] for _ in inputs] |
| 60 | |
| 61 | |
| 62 | class DummyText2TextGenerationPipeline: |