From 8247915345658abd6ee939486eac23bdddafb717 Mon Sep 17 00:00:00 2001 From: Anthony Mercurio Date: Sat, 29 Oct 2022 16:11:48 -0700 Subject: [PATCH] add attribution --- diffusers_trainer.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/diffusers_trainer.py b/diffusers_trainer.py index b780341..f1c99aa 100644 --- a/diffusers_trainer.py +++ b/diffusers_trainer.py @@ -145,6 +145,12 @@ class ImageStore: with open(filename, 'r') as f: return f.read() + +# ====================================== # +# Bucketing code stolen from hasuwoof: # +# https://github.com/hasuwoof/huskystack # +# ====================================== # + class AspectBucket: def __init__(self, store: ImageStore, num_buckets: int,