Support latent previews for nested latents (#15196)

This commit is contained in:
Jukka Seppänen
2026-08-02 02:57:15 +03:00
committed by GitHub
parent e803f24ea0
commit 49a7422892
5 changed files with 153 additions and 8 deletions

View File

@@ -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))