Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(nodes): use LATENT_SCALE_FACTOR in primitives.py, noise.py

Browse Source 
- LatentsOutput.build
- NoiseOutput.build
- Noise.width, Noise.height multiple_of
This commit is contained in:
psychedelicious
2024-02-11 08:54:01 +11:00
parent e5d8921cf2
commit e0694a2856
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/invocations/primitives.py
View File

@ -4,6 +4,7 @@ from typing import Optional
import torch
from invokeai.app.invocations.constants import LATENT_SCALE_FACTOR
from invokeai.app.invocations.fields import (
ColorField,
ConditioningField,
@ -321,8 +322,8 @@ class LatentsOutput(BaseInvocationOutput):
def build(cls, latents_name: str, latents: torch.Tensor, seed: Optional[int] = None) -> "LatentsOutput":
return cls(
latents=LatentsField(latents_name=latents_name, seed=seed),
width=latents.size()[3] * 8,
height=latents.size()[2] * 8,
width=latents.size()[3] * LATENT_SCALE_FACTOR,
height=latents.size()[2] * LATENT_SCALE_FACTOR,
)