Convert the torso of the model to float16.
(self)
| 803 | |
| 804 | |
| 805 | def convert_to_fp16(self): |
| 806 | """ |
| 807 | Convert the torso of the model to float16. |
| 808 | """ |
| 809 | self.input_blocks.apply(convert_module_to_f16) |
| 810 | self.middle_block.apply(convert_module_to_f16) |
| 811 | self.output_blocks.apply(convert_module_to_f16) |
| 812 | |
| 813 | def convert_to_fp32(self): |
| 814 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected