Upload folder using huggingface_hub
Browse files- gpt2_dna_translation.py +0 -2
gpt2_dna_translation.py
CHANGED
|
@@ -126,8 +126,6 @@ class GPT2DNATranslationPipeline(Pipeline):
|
|
| 126 |
return ModelOutput({"sequences": output_ids})
|
| 127 |
|
| 128 |
def postprocess(self, model_outputs: dict, **kwargs):
|
| 129 |
-
assert isinstance(self.tokenizer, GPT2Tokenizer)
|
| 130 |
-
|
| 131 |
output_ids = model_outputs["sequences"]
|
| 132 |
generated = self.tokenizer.decode(output_ids[0])
|
| 133 |
|
|
|
|
| 126 |
return ModelOutput({"sequences": output_ids})
|
| 127 |
|
| 128 |
def postprocess(self, model_outputs: dict, **kwargs):
|
|
|
|
|
|
|
| 129 |
output_ids = model_outputs["sequences"]
|
| 130 |
generated = self.tokenizer.decode(output_ids[0])
|
| 131 |
|