Refactor model loading

2025-08-16 16:03:15 +03:00
parent e5b0b53265
commit 7fc5ec305d
7 changed files with 343 additions and 227 deletions
@@ -173,7 +173,7 @@ def apply_lora(model, device_to, transformer_load_device, params_to_keep=None, d
                to_load.append((n, m, params))

        to_load.sort(reverse=True)
-        pbar = ProgressBar(len(to_load))
+        #pbar = ProgressBar(len(to_load))
        for x in tqdm(to_load, desc="Loading model and applying LoRA weights:", leave=True):
            name = x[0]
            m = x[1]
@@ -207,7 +207,7 @@ def apply_lora(model, device_to, transformer_load_device, params_to_keep=None, d
                    except:
                        continue
            m.comfy_patched_weights = True
-            pbar.update(1)
+            #pbar.update(1)
      
        model.current_weight_patches_uuid = model.patches_uuid
        if low_mem_load: