Jacob Renn
commited on
Commit
·
ecf8b7a
1
Parent(s):
f9a8520
updating pipeline
Browse filesSigned-off-by: Jacob Renn <77127228+jacobrenn@users.noreply.github.com>
- instruct_pipeline.py +1 -0
instruct_pipeline.py
CHANGED
|
@@ -111,6 +111,7 @@ class InstructionTextGenerationPipeline(Pipeline):
|
|
| 111 |
generated_sequence = model_outputs["generated_sequence"][0]
|
| 112 |
instruction_text = model_outputs["instruction_text"]
|
| 113 |
|
|
|
|
| 114 |
records = []
|
| 115 |
for sequence in generated_sequence:
|
| 116 |
|
|
|
|
| 111 |
generated_sequence = model_outputs["generated_sequence"][0]
|
| 112 |
instruction_text = model_outputs["instruction_text"]
|
| 113 |
|
| 114 |
+
generated_sequence = generated_sequence.numpy().tolist()
|
| 115 |
records = []
|
| 116 |
for sequence in generated_sequence:
|
| 117 |
|