|
16 | 16 |
|
17 | 17 | from ...models import MODEL_NAME_TO_MODEL
|
18 | 18 | from ...models.model_factory import EagerModelFactory
|
19 |
| -from ...portable.utils import export_to_exec_prog, save_pte_program |
| 19 | +from ...portable.utils import export_to_exec_prog |
20 | 20 |
|
21 | 21 |
|
22 | 22 | def save_bundled_program(
|
@@ -54,9 +54,8 @@ def save_bundled_program(
|
54 | 54 | file.write(bundled_program_buffer)
|
55 | 55 |
|
56 | 56 |
|
57 |
| -def export_to_pte(model_name, model, method_names, example_inputs): |
| 57 | +def export_to_bundled_program(model_name, model, method_names, example_inputs): |
58 | 58 | exec_prog = export_to_exec_prog(model, example_inputs)
|
59 |
| - save_pte_program(exec_prog.buffer, model_name) |
60 | 59 |
|
61 | 60 | # Just as an example to show how multiple input sets can be bundled along, here we
|
62 | 61 | # create a list with the example_inputs tuple used twice. Each instance of example_inputs
|
@@ -91,4 +90,4 @@ def export_to_pte(model_name, model, method_names, example_inputs):
|
91 | 90 |
|
92 | 91 | method_names = ["forward"]
|
93 | 92 |
|
94 |
| - export_to_pte(args.model_name, model, method_names, example_inputs) |
| 93 | + export_to_bundled_program(args.model_name, model, method_names, example_inputs) |
0 commit comments