Convert the torso of the model to float16.
(self)
| 738 | ) |
| 739 | |
| 740 | def convert_to_fp16(self): |
| 741 | """ |
| 742 | Convert the torso of the model to float16. |
| 743 | """ |
| 744 | self.input_blocks.apply(convert_module_to_f16) |
| 745 | self.middle_block.apply(convert_module_to_f16) |
| 746 | self.output_blocks.apply(convert_module_to_f16) |
| 747 | |
| 748 | def convert_to_fp32(self): |
| 749 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected