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

Adjusted some names and allow to ignore text encoder on extract

Browse Source
This commit is contained in:
Jaret Burkett
2023-07-16 12:52:39 -06:00
parent a61914ddd9
commit 9df7af1720
5 changed files with 39 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jobs/process/BaseExtractProcess.py
View File

@@ -30,6 +30,8 @@ class BaseExtractProcess(BaseProcess):
self.config = config
self.dtype = self.get_conf('dtype', self.job.dtype)
self.torch_dtype = get_torch_dtype(self.dtype)
self.extract_unet = self.get_conf('extract_unet', self.job.extract_unet)
self.extract_text_encoder = self.get_conf('extract_text_encoder', self.job.extract_text_encoder)
def run(self):
# here instead of init because child init needs to go first