From 1120bccc23917f3b86509d4defc3a193cf13d7ad Mon Sep 17 00:00:00 2001
From: Rattus <rattus128@gmail.com>
Date: Sat, 14 Mar 2026 09:15:57 +1000
Subject: [PATCH] ops: put meta-tensors in place to allow custom nodes to check
 geo

---
 comfy/ops.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/comfy/ops.py b/comfy/ops.py
index 077c73b61..3f2da4e63 100644
--- a/comfy/ops.py
+++ b/comfy/ops.py
@@ -578,7 +578,12 @@ class disable_weight_init:
             self.norm_type = norm_type
             self.scale_grad_by_freq = scale_grad_by_freq
             self.sparse = sparse
-            self.weight = None
+            # Keep shape/dtype visible for module introspection without reserving storage.
+            embedding_dtype = dtype if dtype is not None else torch.get_default_dtype()
+            self.weight = torch.nn.Parameter(
+                torch.empty((num_embeddings, embedding_dim), device="meta", dtype=embedding_dtype),
+                requires_grad=False,
+            )
             self.bias = None
             self.weight_comfy_model_dtype = dtype