(all_segments, seg_texts, n_files)
| 80 | |
| 81 | |
| 82 | def concat_results(all_segments, seg_texts, n_files): |
| 83 | file_texts = {} |
| 84 | for (fi, _), text in zip(all_segments, seg_texts): |
| 85 | file_texts.setdefault(fi, []).append(text) |
| 86 | return ["".join(file_texts.get(fi, [])) for fi in range(n_files)] |
| 87 | |
| 88 | |
| 89 | def run_pytorch(model_name, seg_files, device="cuda:0"): |
no outgoing calls
no test coverage detected
searching dependent graphs…