mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-07-21 21:01:24 +08:00
14 lines
387 B
Python
14 lines
387 B
Python
import torch
|
|
|
|
from modules_forge.packages.huggingface_guess.latent import LatentFormat
|
|
|
|
|
|
class ProcessLatent:
|
|
latent_format: LatentFormat = None
|
|
|
|
def process_in(self, latent: torch.Tensor) -> torch.Tensor:
|
|
return self.latent_format.process_in(latent)
|
|
|
|
def process_out(self, latent: torch.Tensor) -> torch.Tensor:
|
|
return self.latent_format.process_out(latent)
|