Convert the torso of the model to float32.
(self)
| 746 | self.output_blocks.apply(convert_module_to_f16) |
| 747 | |
| 748 | def convert_to_fp32(self): |
| 749 | """ |
| 750 | Convert the torso of the model to float32. |
| 751 | """ |
| 752 | self.input_blocks.apply(convert_module_to_f32) |
| 753 | self.middle_block.apply(convert_module_to_f32) |
| 754 | self.output_blocks.apply(convert_module_to_f32) |
| 755 | |
| 756 | def forward(self, x, timesteps=None, context=None, y=None,**kwargs): |
| 757 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected