refactor devices

Signed-off-by: Vladimir Mandic <mandic00@live.com>
2026-09-20 01:31:13 +02:00 · 2024-09-29 20:17:03 -04:00
parent fe94edf781
commit 47755dce6b
14 changed files with 182 additions and 182 deletions
@@ -1737,6 +1737,7 @@ def reload_text_encoder(initial=False):


 def reload_model_weights(sd_model=None, info=None, reuse_dict=False, op='model', force=False):
+    devices.set_cuda_params()
    load_dict = shared.opts.sd_model_dict != model_data.sd_dict
    from modules import lowvram, sd_hijack
    checkpoint_info = info or select_checkpoint(op=op) # are we selecting model or dictionary