mirror of
https://github.com/Comfy-Org/ComfyUI.git
synced 2026-08-17 23:08:53 +08:00
Support latent previews for nested latents (#15196)
This commit is contained in:
@@ -51,6 +51,9 @@ class NestedTensor:
|
||||
def float(self):
|
||||
return self.to(dtype=torch.float)
|
||||
|
||||
def cpu(self):
|
||||
return self.to(device="cpu")
|
||||
|
||||
def chunk(self, *args, **kwargs):
|
||||
return self.apply_operation(None, lambda x, y: x.chunk(*args, **kwargs))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user