From 1a82e4b48fa697cb7a3cdc28e85c16710f0c5db7 Mon Sep 17 00:00:00 2001
From: aszc-dev <kontakt@aszczepanski.pl>
Date: Thu, 16 Nov 2023 00:05:09 +0100
Subject: [PATCH] Remove CLIP related code

---
 coreml_suite/clip.py | 22 ----------------------
 1 file changed, 22 deletions(-)
 delete mode 100644 coreml_suite/clip.py

diff --git a/coreml_suite/clip.py b/coreml_suite/clip.py
deleted file mode 100644
index 8a46b51..0000000
--- a/coreml_suite/clip.py
+++ /dev/null
@@ -1,22 +0,0 @@
-import numpy as np
-import torch
-
-from comfy.sd import CLIP
-from comfy.sd1_clip import SDClipModel
-
-
-class CoreMLCLIP(CLIP):
-    pass
-
-
-class SDClipModelCoreML(SDClipModel):
-    def __init__(self, **kwargs):
-        self.model = kwargs.pop("coreml_model")
-        super().__init__(**kwargs)
-
-    def encode_token_weights(self, token_weight_pairs):
-        tokens = np.array(token_weight_pairs["l"]).astype(np.float16)[:, :, 0]
-        encoded = self.model(input_ids=tokens)
-        cond = torch.from_numpy(encoded["last_hidden_state"])
-        pooled = torch.from_numpy(encoded["pooled_outputs"])
-        return cond, pooled