(model)
| 112 | |
| 113 | |
| 114 | def unload_model(model): |
| 115 | global lowvram_mode |
| 116 | if lowvram_mode: |
| 117 | model.cpu() |
| 118 | torch.cuda.empty_cache() |
| 119 | |
| 120 | |
| 121 | def load_model_from_config(config, ckpt=None, verbose=True): |
no outgoing calls
no test coverage detected