Update the fake exported program to point to the real state dict. Modifies the fake exported program in-place.
(
fake_exported_program: ExportedProgram, real_exported_program: ExportedProgram
)
| 58 | |
| 59 | |
| 60 | def update_to_real_program( |
| 61 | fake_exported_program: ExportedProgram, real_exported_program: ExportedProgram |
| 62 | ) -> None: |
| 63 | """Update the fake exported program to point to the real state dict. Modifies the |
| 64 | fake exported program in-place. |
| 65 | """ |
| 66 | for k, v in real_exported_program.state_dict.items(): |
| 67 | fake_exported_program._state_dict[k] = v |