File size: 214 Bytes
82ea528 |
1 2 3 4 5 6 7 8 9 10 |
import torch
class LatentGuide(torch.nn.Module):
def __init__(self, latent: torch.Tensor, index) -> None:
super().__init__()
self.index = index
self.register_buffer('latent', latent)
|