WIP porting to kohya-sdxl. So much to do.

2023-08-15 17:07:34 -06:00
parent 55a5fcc7d9
commit 1d2523b978
5 changed files with 328 additions and 58 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -76,6 +76,7 @@ class ModelConfig:
        self.is_xl: bool = kwargs.get('is_xl', False)
        self.is_v_pred: bool = kwargs.get('is_v_pred', False)
        self.dtype: str = kwargs.get('dtype', 'float16')
+        self.vae_path: str = kwargs.get('vae_path', None)

        if self.name_or_path is None:
            raise ValueError('name_or_path must be specified')