mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
21 lines
886 B
Python
21 lines
886 B
Python
from typing import Literal
|
|
|
|
from invokeai.backend.stable_diffusion.schedulers import SCHEDULER_MAP
|
|
from invokeai.backend.util.devices import TorchDevice
|
|
|
|
LATENT_SCALE_FACTOR = 8
|
|
"""
|
|
HACK: Many nodes are currently hard-coded to use a fixed latent scale factor of 8. This is fragile, and will need to
|
|
be addressed if future models use a different latent scale factor. Also, note that there may be places where the scale
|
|
factor is hard-coded to a literal '8' rather than using this constant.
|
|
The ratio of image:latent dimensions is LATENT_SCALE_FACTOR:1, or 8:1.
|
|
"""
|
|
|
|
SCHEDULER_NAME_VALUES = Literal[tuple(SCHEDULER_MAP.keys())]
|
|
"""A literal type representing the valid scheduler names."""
|
|
|
|
IMAGE_MODES = Literal["L", "RGB", "RGBA", "CMYK", "YCbCr", "LAB", "HSV", "I", "F"]
|
|
"""A literal type for PIL image modes supported by Invoke"""
|
|
|
|
DEFAULT_PRECISION = TorchDevice.choose_torch_dtype()
|