ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-27 17:51:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:ostris/ai-toolkit

Browse Source
This commit is contained in:
Jaret Burkett
2024-06-23 14:47:43 -06:00
parent 64f2b085b7 7165f2d25a
commit 8407c4deea
6 changed files with 65 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/config_modules.py
View File

@@ -372,6 +372,8 @@ class ModelConfig:
# for text encoder quant. Only works with pixart currently
self.text_encoder_bits = kwargs.get('text_encoder_bits', 8) # 16, 8, 4
self.unet_path = kwargs.get("unet_path", None)
self.unet_sample_size = kwargs.get("unet_sample_size", None)
class ReferenceDatasetConfig: