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

Added adapter modules for text encoders and direct vision

Browse Source
This commit is contained in:
Jaret Burkett
2024-02-12 08:46:18 -07:00
parent e074058faa
commit 4ec4025cbb
7 changed files with 645 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -698,7 +698,8 @@ class SDTrainer(BaseSDTrainProcess):
can_disable_adapter = False
was_adapter_active = False
if self.adapter is not None and (isinstance(self.adapter, IPAdapter) or
isinstance(self.adapter, ReferenceAdapter)
isinstance(self.adapter, ReferenceAdapter) or
(isinstance(self.adapter, CustomAdapter))
):
can_disable_adapter = True
was_adapter_active = self.adapter.is_active