rework several component patcher

backend is 65% finished
2024-08-02 14:56:44 -07:00
parent 91543adb90
commit 6ed95d66e5
8 changed files with 461 additions and 235 deletions
@@ -1,14 +0,0 @@
-import torch
-
-
-class JointTokenizer:
-    def __init__(self, huggingface_components):
-        self.clip_l = huggingface_components.get('tokenizer', None)
-        self.clip_g = huggingface_components.get('tokenizer_2', None)
-
-
-class JointCLIP(torch.nn.Module):
-    def __init__(self, huggingface_components):
-        super().__init__()
-        self.clip_l = huggingface_components.get('text_encoder', None)
-        self.clip_g = huggingface_components.get('text_encoder_2', None)