2024-02-10 22:51:25 +00:00
|
|
|
from typing import Literal
|
|
|
|
|
|
|
|
|
|
from invokeai.backend.stable_diffusion.schedulers import SCHEDULER_MAP
|
|
|
|
|
|
2024-02-10 21:52:07 +00:00
|
|
|
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.
|
|
|
|
|
"""
|
2024-02-10 22:51:25 +00:00
|
|
|
|
|
|
|
|
SCHEDULER_NAME_VALUES = Literal[tuple(SCHEDULER_MAP.keys())]
|
|
|
|
|
"""A literal type representing the valid scheduler names."""
|
2024-02-18 21:56:46 +00:00
|
|
|
|
|
|
|
|
IMAGE_MODES = Literal["L", "RGB", "RGBA", "CMYK", "YCbCr", "LAB", "HSV", "I", "F"]
|
|
|
|
|
"""A literal type for PIL image modes supported by Invoke"""
|
