From 3d62fa959883bd7358cc548bb945113b8a7f323f Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Thu, 29 Aug 2024 20:17:32 -0700 Subject: [PATCH] reduce prints --- backend/patcher/lora.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/backend/patcher/lora.py b/backend/patcher/lora.py index 5d6083a2..6e0e05d7 100644 --- a/backend/patcher/lora.py +++ b/backend/patcher/lora.py @@ -336,8 +336,6 @@ class LoraLoader: self.dirty = False - execution_start_time = time.perf_counter() - # Initialize memory_management.signal_empty_cache = True @@ -426,16 +424,14 @@ class LoraLoader: utils.set_attr_raw(self.model, key, torch.nn.Parameter(weight, requires_grad=False)) - # Time + # End set_parameter_devices(self.model, parameter_devices=parameter_devices) - moving_time = time.perf_counter() - execution_start_time - if len(self.patches) > 0: if self.online_mode: - print(f'Patching LoRA on-the-fly in {moving_time:.2f} seconds.') + print(f'Patched LoRAs on-the-fly; ', end='') else: - print(f'Patching LoRA by precomputing model weights in {moving_time:.2f} seconds.') + print(f'Patched LoRAs by precomputing model weights; ', end='') return