(Initial LTXV2 DualCLIPLoader Support) Update TensorEncoder implementation and enhance tensor handling; bump version to 1.1.1

This commit is contained in:
nyueki
2026-02-16 19:46:13 -05:00
parent 72df1e9d80
commit ef3bda2be9
2 changed files with 6 additions and 11 deletions
+5 -10
View File
@@ -8,12 +8,6 @@ import folder_paths
import comfy.sd
import comfy.utils
class TensorEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, torch.Tensor):
return obj.cpu().tolist()
return super().default(obj)
class TensorEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, torch.Tensor):
@@ -59,7 +53,7 @@ def pack_tensors(tensors):
t = torch.zeros((1, 77, 768))
else:
t = torch.zeros((1, 768))
t = t.contiguous().cpu()
t = t.detach().contiguous().cpu()
raw = t.numpy().tobytes()
meta[name] = {"dtype": str(t.dtype), "shape": list(t.shape), "size": len(raw)}
blobs.append(raw)
@@ -146,9 +140,10 @@ class SendRemoteCLIP:
log("Warning: invalid request, skipping")
continue
log(f"Encoding prompt ({len(meta['text'])} chars)")
tokens = CLIP.tokenize(meta["text"], **meta["kwargs"])
cond, pooled = CLIP.encode_from_tokens(tokens, return_pooled=True)
tensor_meta, blob = pack_tensors({"cond": cond, "pooled": pooled})
with torch.inference_mode():
tokens = CLIP.tokenize(meta["text"], **meta["kwargs"])
cond, pooled = CLIP.encode_from_tokens(tokens, return_pooled=True)
tensor_meta, blob = pack_tensors({"cond": cond, "pooled": pooled})
reply_meta = {"tensors": tensor_meta, "blob_size": len(blob)}
send_packet(conn, reply_meta, blob)
log(f"Sent embeddings ({len(blob)} bytes)")
+1 -1
View File
@@ -1,7 +1,7 @@
[project]
name = "remotecliploader"
description = ""
version = "1.1.0"
version = "1.1.1"
license = {file = "LICENSE"}
# classifiers = [
# # For OS-independent nodes (works on all operating systems)