Switched to new bucket system that matched sdxl trained buckets. Fixed requirements. Updated embeddings to work with sdxl. Added method to train lora with an embedding at the trigger. Still testing but works amazingly well from what I can see

toolkit/buckets.py

@@ -1,6 +1,10 @@
-from typing import Type, List, Union
+from typing import Type, List, Union, TypedDict
+
+
+class BucketResolution(TypedDict):
+    width: int
+    height: int
 
-BucketResolution = Type[{"width": int, "height": int}]
 
 # resolutions SDXL was trained on with a 1024x1024 base resolution
 resolutions_1024: List[BucketResolution] = [