()
| 69 | |
| 70 | |
| 71 | def reload_sd_model(): |
| 72 | from modules import shared, devices |
| 73 | if shared.sd_model is not None: |
| 74 | if shared.sd_model.cond_stage_model is not None: |
| 75 | shared.sd_model.cond_stage_model.to(devices.device) |
| 76 | if shared.sd_model.first_stage_model: |
| 77 | shared.sd_model.first_stage_model.to(devices.device) |
| 78 | # Maybe something else??? |
| 79 | |
| 80 | def get_hide_dirs(): |
| 81 | import modules.shared |