mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
29 lines
661 B
Python
29 lines
661 B
Python
|
# TODO: Make a new model for this
|
||
|
|
from enum import Enum
|
||
|
|
|
||
|
|
from invokeai.app.util.enum import MetaEnum
|
||
|
|
|
||
|
|
|
||
|
|
class ResourceType(str, Enum, metaclass=MetaEnum):
|
||
|
|
"""The type of a resource."""
|
||
|
|
|
||
|
|
IMAGES = "images"
|
||
|
|
TENSORS = "tensors"
|
||
|
|
|
||
|
|
|
||
|
|
# class ResourceOrigin(str, Enum, metaclass=MetaEnum):
|
||
|
|
# """The origin of a resource (eg image or tensor)."""
|
||
|
|
|
||
|
|
# RESULTS = "results"
|
||
|
|
# UPLOADS = "uploads"
|
||
|
|
# INTERMEDIATES = "intermediates"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class TensorKind(str, Enum, metaclass=MetaEnum):
|
||
|
|
"""The kind of a tensor. Use TensorKind.OTHER for non-default kinds."""
|
||
|
|
|
||
|
|
IMAGE_LATENTS = "image_latents"
|
||
|
|
CONDITIONING = "conditioning"
|
||
|
|
OTHER = "other"
|