Added support to disable single transformers in vision direct adapter

2024-09-11 08:54:51 -06:00
parent fc34a69bec
commit 951e223481
2 changed files with 23 additions and 12 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -203,6 +203,8 @@ class AdapterConfig:
        self.ilora_mid: bool = kwargs.get('ilora_mid', True)
        self.ilora_up: bool = kwargs.get('ilora_up', True)

+        self.flux_only_double: bool = kwargs.get('flux_only_double', False)
+

 class EmbeddingConfig:
    def __init__(self, **kwargs):