Convert the torso of the model to float16.
(self)
| 616 | ) |
| 617 | |
| 618 | def convert_to_fp16(self): |
| 619 | """ |
| 620 | Convert the torso of the model to float16. |
| 621 | """ |
| 622 | self.input_blocks.apply(convert_module_to_f16) |
| 623 | self.middle_block.apply(convert_module_to_f16) |
| 624 | self.output_blocks.apply(convert_module_to_f16) |
| 625 | |
| 626 | def convert_to_fp32(self): |
| 627 | """ |