()
| 59 | |
| 60 | |
| 61 | def unload_sd_model(): |
| 62 | from modules import shared, devices |
| 63 | if shared.sd_model is not None: |
| 64 | if shared.sd_model.cond_stage_model is not None: |
| 65 | shared.sd_model.cond_stage_model.to(devices.cpu) |
| 66 | if shared.sd_model.first_stage_model is not None: |
| 67 | shared.sd_model.first_stage_model.to(devices.cpu) |
| 68 | # Maybe something else??? |
| 69 | |
| 70 | |
| 71 | def reload_sd_model(): |