Remove references to model_records service, change submodel property on ModelInfo to submodel_type to support new params in model manager

This commit is contained in:
Brandon Rising
2024-02-14 09:36:30 -05:00
committed by psychedelicious
parent 3e82f63c7e
commit 86ac55ab5f
6 changed files with 29 additions and 29 deletions

View File

@ -812,7 +812,7 @@ class LatentsToImageInvocation(BaseInvocation, WithMetadata, WithBoard):
)
with set_seamless(vae_info.model, self.vae.seamless_axes), vae_info as vae:
assert isinstance(vae, torch.Tensor)
assert isinstance(vae, torch.nn.Module)
latents = latents.to(vae.device)
if self.fp32:
vae.to(dtype=torch.float32)