better prints

This commit is contained in:
layerdiffusion
2024-08-16 21:13:09 -07:00
parent f3e211d431
commit 9973d5dc09

View File

@@ -303,7 +303,7 @@ class LoraLoader:
# Patch
for key, current_patches in (tqdm(self.patches.items(), desc='Patching LoRAs') if len(self.patches) > 0 else self.patches):
for key, current_patches in (tqdm(self.patches.items(), desc=f'Patching LoRAs for {type(self.model).__name__}') if len(self.patches) > 0 else self.patches):
try:
parent_layer, weight = utils.get_attr_with_parent(self.model, key)
assert isinstance(weight, torch.nn.Parameter)