Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add attribution

This commit is contained in:
Anthony Mercurio 2022-10-29 16:11:48 -07:00 committed by GitHub
parent be702cce2e
commit 8247915345
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
diffusers_trainer.py
View File

@ -145,6 +145,12 @@ class ImageStore:
with open(filename, 'r') as f: with open(filename, 'r') as f:
return f.read() return f.read()
# ====================================== #
# Bucketing code stolen from hasuwoof: #
# https://github.com/hasuwoof/huskystack #
# ====================================== #
class AspectBucket: class AspectBucket:
def __init__(self, store: ImageStore, def __init__(self, store: ImageStore,
num_buckets: int, num_buckets: int,