Initial training script for photomaker training. Needs a little more work.

toolkit/config_modules.py

@@ -127,7 +127,7 @@ class NetworkConfig:
                 self.conv = 4
 
 
-AdapterTypes = Literal['t2i', 'ip', 'ip+', 'clip', 'ilora']
+AdapterTypes = Literal['t2i', 'ip', 'ip+', 'clip', 'ilora', 'photo_maker']
 
 
 class AdapterConfig:
@@ -162,6 +162,8 @@ class AdapterConfig:
         self.trigger = kwargs.get('trigger', 'tri993r')
         self.trigger_class_name = kwargs.get('trigger_class_name', 'person')
 
+        self.class_names = kwargs.get('class_names', [])
+
 
 class EmbeddingConfig:
     def __init__(self, **kwargs):