mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-28 18:21:16 +00:00
Added adapter modules for text encoders and direct vision
This commit is contained in:
@@ -177,6 +177,10 @@ class AdapterConfig:
|
||||
else:
|
||||
self.clip_layer = 'last_hidden_state'
|
||||
|
||||
# text encoder
|
||||
self.text_encoder_path: str = kwargs.get('text_encoder_path', None)
|
||||
self.text_encoder_arch: str = kwargs.get('text_encoder_arch', 'clip') # clip t5
|
||||
|
||||
|
||||
class EmbeddingConfig:
|
||||
def __init__(self, **kwargs):
|
||||
|
||||
Reference in New Issue
Block a user