hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Hypernetworks - fix KeyError in statistics caching

Browse Source 
Statistics logging has changed to {filename : list[losses]}, so it has to use loss_info[key].pop()
This commit is contained in:
AngelBottomless
2022-10-23 20:06:42 +09:00
committed by AUTOMATIC1111
parent f2cc3f32d5
commit b297cc3324
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/hypernetworks/hypernetwork.py
View File

@@ -274,8 +274,8 @@ def log_statistics(loss_info:dict, key, value):
loss_info[key] = [value]
else:
loss_info[key].append(value)
if len(loss_info) > 1024:
loss_info.pop(0)
if len(loss_info[key]) > 1024:
loss_info[key].pop(0)
def statistics(data):