Added support for FLUX.1-Kontext-dev
This commit is contained in:
1 parent
8d9c47316a
commit
60ef2f1df7
12 files changed
+570
-4
No files matched your search
@@ -757,6 +757,8 @@ class DatasetConfig:
|
||||
self.flip_y: bool = kwargs.get('flip_y', False)
|
||||
self.augments: List[str] = kwargs.get('augments', [])
|
||||
self.control_path: Union[str,List[str]] = kwargs.get('control_path', None) # depth maps, etc
|
||||
if self.control_path == '':
|
||||
self.control_path = None
|
||||
# inpaint images should be webp/png images with alpha channel. The alpha 0 (invisible) section will
|
||||
# be the part conditioned to be inpainted. The alpha 1 (visible) section will be the part that is ignored
|
||||
self.inpaint_path: Union[str,List[str]] = kwargs.get('inpaint_path', None)
|
||||
|
||||
Reference in new issue
Block a user