From bd47167526e59c9cf3fdd8e8de364dbae96550df Mon Sep 17 00:00:00 2001
From: aria1th <35677394+aria1th@users.noreply.github.com>
Date: Fri, 28 Jul 2023 21:18:50 +0900
Subject: [PATCH] fix missing report_statistics in 1.5.1

fixes https://github.com/aria1th/Hypernetwork-MonkeyPatch-Extension/issues/54
---
 patches/external_pr/hypernetwork.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/patches/external_pr/hypernetwork.py b/patches/external_pr/hypernetwork.py
index 9fd6da7..47f072c 100644
--- a/patches/external_pr/hypernetwork.py
+++ b/patches/external_pr/hypernetwork.py
@@ -12,7 +12,7 @@ import torch
 import tqdm
 
 from modules import shared, sd_models, devices, processing, sd_samplers
-from modules.hypernetworks.hypernetwork import optimizer_dict, stack_conds, save_hypernetwork, report_statistics
+from modules.hypernetworks.hypernetwork import optimizer_dict, stack_conds, save_hypernetwork
 from modules.textual_inversion import textual_inversion
 from modules.textual_inversion.learn_schedule import LearnRateScheduler
 from ..tbutils import tensorboard_setup, tensorboard_add, tensorboard_add_image, tensorboard_log_hyperparameter
@@ -560,7 +560,6 @@ Last saved image: {html.escape(last_saved_image)}<br/>
         remove_accessible()
         gc.collect()
         torch.cuda.empty_cache()
-    report_statistics(loss_dict)
     filename = os.path.join(shared.cmd_opts.hypernetwork_dir, f'{hypernetwork_name}.pt')
     hypernetwork.optimizer_name = optimizer_name
     if shared.opts.save_optimizer_state:
@@ -1115,7 +1114,6 @@ Last saved image: {html.escape(last_saved_image)}<br/>
                                            loss=mean_loss,
                                            base_hypernetwork_name=hypernetwork_name
                                            )
-    report_statistics(loss_dict)
     filename = os.path.join(shared.cmd_opts.hypernetwork_dir, f'{hypernetwork_name}.pt')
     hypernetwork.optimizer_name = optimizer_name
     if shared.opts.save_optimizer_state: