diff --git a/backend/patcher/lora.py b/backend/patcher/lora.py index 51e38494..238744b2 100644 --- a/backend/patcher/lora.py +++ b/backend/patcher/lora.py @@ -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)