mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
Add ruff rule to disallow relative parent imports.
This commit is contained in:
parent
7bbd793064
commit
9da5925287
@ -4,37 +4,39 @@ from logging import Logger
|
|||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
|
from invokeai.app.services.board_image_records.board_image_records_sqlite import SqliteBoardImageRecordStorage
|
||||||
|
from invokeai.app.services.board_images.board_images_default import BoardImagesService
|
||||||
|
from invokeai.app.services.board_records.board_records_sqlite import SqliteBoardRecordStorage
|
||||||
|
from invokeai.app.services.boards.boards_default import BoardService
|
||||||
|
from invokeai.app.services.bulk_download.bulk_download_default import BulkDownloadService
|
||||||
|
from invokeai.app.services.config.config_default import InvokeAIAppConfig
|
||||||
|
from invokeai.app.services.download.download_default import DownloadQueueService
|
||||||
|
from invokeai.app.services.events.events_fastapievents import FastAPIEventService
|
||||||
|
from invokeai.app.services.image_files.image_files_disk import DiskImageFileStorage
|
||||||
|
from invokeai.app.services.image_records.image_records_sqlite import SqliteImageRecordStorage
|
||||||
|
from invokeai.app.services.images.images_default import ImageService
|
||||||
|
from invokeai.app.services.invocation_cache.invocation_cache_memory import MemoryInvocationCache
|
||||||
|
from invokeai.app.services.invocation_services import InvocationServices
|
||||||
|
from invokeai.app.services.invocation_stats.invocation_stats_default import InvocationStatsService
|
||||||
|
from invokeai.app.services.invoker import Invoker
|
||||||
|
from invokeai.app.services.model_images.model_images_default import ModelImageFileStorageDisk
|
||||||
|
from invokeai.app.services.model_manager.model_manager_default import ModelManagerService
|
||||||
|
from invokeai.app.services.model_records.model_records_sql import ModelRecordServiceSQL
|
||||||
|
from invokeai.app.services.names.names_default import SimpleNameService
|
||||||
from invokeai.app.services.object_serializer.object_serializer_disk import ObjectSerializerDisk
|
from invokeai.app.services.object_serializer.object_serializer_disk import ObjectSerializerDisk
|
||||||
from invokeai.app.services.object_serializer.object_serializer_forward_cache import ObjectSerializerForwardCache
|
from invokeai.app.services.object_serializer.object_serializer_forward_cache import ObjectSerializerForwardCache
|
||||||
|
from invokeai.app.services.session_processor.session_processor_default import (
|
||||||
|
DefaultSessionProcessor,
|
||||||
|
DefaultSessionRunner,
|
||||||
|
)
|
||||||
|
from invokeai.app.services.session_queue.session_queue_sqlite import SqliteSessionQueue
|
||||||
from invokeai.app.services.shared.sqlite.sqlite_util import init_db
|
from invokeai.app.services.shared.sqlite.sqlite_util import init_db
|
||||||
|
from invokeai.app.services.urls.urls_default import LocalUrlService
|
||||||
|
from invokeai.app.services.workflow_records.workflow_records_sqlite import SqliteWorkflowRecordsStorage
|
||||||
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import ConditioningFieldData
|
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import ConditioningFieldData
|
||||||
from invokeai.backend.util.logging import InvokeAILogger
|
from invokeai.backend.util.logging import InvokeAILogger
|
||||||
from invokeai.version.invokeai_version import __version__
|
from invokeai.version.invokeai_version import __version__
|
||||||
|
|
||||||
from ..services.board_image_records.board_image_records_sqlite import SqliteBoardImageRecordStorage
|
|
||||||
from ..services.board_images.board_images_default import BoardImagesService
|
|
||||||
from ..services.board_records.board_records_sqlite import SqliteBoardRecordStorage
|
|
||||||
from ..services.boards.boards_default import BoardService
|
|
||||||
from ..services.bulk_download.bulk_download_default import BulkDownloadService
|
|
||||||
from ..services.config import InvokeAIAppConfig
|
|
||||||
from ..services.download import DownloadQueueService
|
|
||||||
from ..services.events.events_fastapievents import FastAPIEventService
|
|
||||||
from ..services.image_files.image_files_disk import DiskImageFileStorage
|
|
||||||
from ..services.image_records.image_records_sqlite import SqliteImageRecordStorage
|
|
||||||
from ..services.images.images_default import ImageService
|
|
||||||
from ..services.invocation_cache.invocation_cache_memory import MemoryInvocationCache
|
|
||||||
from ..services.invocation_services import InvocationServices
|
|
||||||
from ..services.invocation_stats.invocation_stats_default import InvocationStatsService
|
|
||||||
from ..services.invoker import Invoker
|
|
||||||
from ..services.model_images.model_images_default import ModelImageFileStorageDisk
|
|
||||||
from ..services.model_manager.model_manager_default import ModelManagerService
|
|
||||||
from ..services.model_records import ModelRecordServiceSQL
|
|
||||||
from ..services.names.names_default import SimpleNameService
|
|
||||||
from ..services.session_processor.session_processor_default import DefaultSessionProcessor, DefaultSessionRunner
|
|
||||||
from ..services.session_queue.session_queue_sqlite import SqliteSessionQueue
|
|
||||||
from ..services.urls.urls_default import LocalUrlService
|
|
||||||
from ..services.workflow_records.workflow_records_sqlite import SqliteWorkflowRecordsStorage
|
|
||||||
|
|
||||||
|
|
||||||
# TODO: is there a better way to achieve this?
|
# TODO: is there a better way to achieve this?
|
||||||
def check_internet() -> bool:
|
def check_internet() -> bool:
|
||||||
|
@ -10,14 +10,13 @@ from fastapi import Body
|
|||||||
from fastapi.routing import APIRouter
|
from fastapi.routing import APIRouter
|
||||||
from pydantic import BaseModel, Field
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
from invokeai.app.api.dependencies import ApiDependencies
|
||||||
from invokeai.app.invocations.upscale import ESRGAN_MODELS
|
from invokeai.app.invocations.upscale import ESRGAN_MODELS
|
||||||
from invokeai.app.services.invocation_cache.invocation_cache_common import InvocationCacheStatus
|
from invokeai.app.services.invocation_cache.invocation_cache_common import InvocationCacheStatus
|
||||||
from invokeai.backend.image_util.infill_methods.patchmatch import PatchMatch
|
from invokeai.backend.image_util.infill_methods.patchmatch import PatchMatch
|
||||||
from invokeai.backend.util.logging import logging
|
from invokeai.backend.util.logging import logging
|
||||||
from invokeai.version import __version__
|
from invokeai.version import __version__
|
||||||
|
|
||||||
from ..dependencies import ApiDependencies
|
|
||||||
|
|
||||||
|
|
||||||
class LogLevel(int, Enum):
|
class LogLevel(int, Enum):
|
||||||
NotSet = logging.NOTSET
|
NotSet = logging.NOTSET
|
||||||
|
@ -2,7 +2,7 @@ from fastapi import Body, HTTPException
|
|||||||
from fastapi.routing import APIRouter
|
from fastapi.routing import APIRouter
|
||||||
from pydantic import BaseModel, Field
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
from ..dependencies import ApiDependencies
|
from invokeai.app.api.dependencies import ApiDependencies
|
||||||
|
|
||||||
board_images_router = APIRouter(prefix="/v1/board_images", tags=["boards"])
|
board_images_router = APIRouter(prefix="/v1/board_images", tags=["boards"])
|
||||||
|
|
||||||
|
@ -4,12 +4,11 @@ from fastapi import Body, HTTPException, Path, Query
|
|||||||
from fastapi.routing import APIRouter
|
from fastapi.routing import APIRouter
|
||||||
from pydantic import BaseModel, Field
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
from invokeai.app.api.dependencies import ApiDependencies
|
||||||
from invokeai.app.services.board_records.board_records_common import BoardChanges
|
from invokeai.app.services.board_records.board_records_common import BoardChanges
|
||||||
from invokeai.app.services.boards.boards_common import BoardDTO
|
from invokeai.app.services.boards.boards_common import BoardDTO
|
||||||
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
|
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
|
||||||
|
|
||||||
from ..dependencies import ApiDependencies
|
|
||||||
|
|
||||||
boards_router = APIRouter(prefix="/v1/boards", tags=["boards"])
|
boards_router = APIRouter(prefix="/v1/boards", tags=["boards"])
|
||||||
|
|
||||||
|
|
||||||
|
@ -8,13 +8,12 @@ from fastapi.routing import APIRouter
|
|||||||
from pydantic.networks import AnyHttpUrl
|
from pydantic.networks import AnyHttpUrl
|
||||||
from starlette.exceptions import HTTPException
|
from starlette.exceptions import HTTPException
|
||||||
|
|
||||||
|
from invokeai.app.api.dependencies import ApiDependencies
|
||||||
from invokeai.app.services.download import (
|
from invokeai.app.services.download import (
|
||||||
DownloadJob,
|
DownloadJob,
|
||||||
UnknownJobIDException,
|
UnknownJobIDException,
|
||||||
)
|
)
|
||||||
|
|
||||||
from ..dependencies import ApiDependencies
|
|
||||||
|
|
||||||
download_queue_router = APIRouter(prefix="/v1/download_queue", tags=["download_queue"])
|
download_queue_router = APIRouter(prefix="/v1/download_queue", tags=["download_queue"])
|
||||||
|
|
||||||
|
|
||||||
|
@ -8,6 +8,7 @@ from fastapi.routing import APIRouter
|
|||||||
from PIL import Image
|
from PIL import Image
|
||||||
from pydantic import BaseModel, Field, JsonValue
|
from pydantic import BaseModel, Field, JsonValue
|
||||||
|
|
||||||
|
from invokeai.app.api.dependencies import ApiDependencies
|
||||||
from invokeai.app.invocations.fields import MetadataField
|
from invokeai.app.invocations.fields import MetadataField
|
||||||
from invokeai.app.services.image_records.image_records_common import (
|
from invokeai.app.services.image_records.image_records_common import (
|
||||||
ImageCategory,
|
ImageCategory,
|
||||||
@ -18,8 +19,6 @@ from invokeai.app.services.images.images_common import ImageDTO, ImageUrlsDTO
|
|||||||
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
|
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
|
||||||
from invokeai.app.services.shared.sqlite.sqlite_common import SQLiteDirection
|
from invokeai.app.services.shared.sqlite.sqlite_common import SQLiteDirection
|
||||||
|
|
||||||
from ..dependencies import ApiDependencies
|
|
||||||
|
|
||||||
images_router = APIRouter(prefix="/v1/images", tags=["images"])
|
images_router = APIRouter(prefix="/v1/images", tags=["images"])
|
||||||
|
|
||||||
|
|
||||||
|
@ -16,6 +16,7 @@ from pydantic import AnyHttpUrl, BaseModel, ConfigDict, Field
|
|||||||
from starlette.exceptions import HTTPException
|
from starlette.exceptions import HTTPException
|
||||||
from typing_extensions import Annotated
|
from typing_extensions import Annotated
|
||||||
|
|
||||||
|
from invokeai.app.api.dependencies import ApiDependencies
|
||||||
from invokeai.app.services.model_images.model_images_common import ModelImageFileNotFoundException
|
from invokeai.app.services.model_images.model_images_common import ModelImageFileNotFoundException
|
||||||
from invokeai.app.services.model_install.model_install_common import ModelInstallJob
|
from invokeai.app.services.model_install.model_install_common import ModelInstallJob
|
||||||
from invokeai.app.services.model_records import (
|
from invokeai.app.services.model_records import (
|
||||||
@ -35,8 +36,6 @@ from invokeai.backend.model_manager.metadata.metadata_base import ModelMetadataW
|
|||||||
from invokeai.backend.model_manager.search import ModelSearch
|
from invokeai.backend.model_manager.search import ModelSearch
|
||||||
from invokeai.backend.model_manager.starter_models import STARTER_MODELS, StarterModel, StarterModelWithoutDependencies
|
from invokeai.backend.model_manager.starter_models import STARTER_MODELS, StarterModel, StarterModelWithoutDependencies
|
||||||
|
|
||||||
from ..dependencies import ApiDependencies
|
|
||||||
|
|
||||||
model_manager_router = APIRouter(prefix="/v2/models", tags=["model_manager"])
|
model_manager_router = APIRouter(prefix="/v2/models", tags=["model_manager"])
|
||||||
|
|
||||||
# images are immutable; set a high max-age
|
# images are immutable; set a high max-age
|
||||||
|
@ -4,6 +4,7 @@ from fastapi import Body, Path, Query
|
|||||||
from fastapi.routing import APIRouter
|
from fastapi.routing import APIRouter
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
from invokeai.app.api.dependencies import ApiDependencies
|
||||||
from invokeai.app.services.session_processor.session_processor_common import SessionProcessorStatus
|
from invokeai.app.services.session_processor.session_processor_common import SessionProcessorStatus
|
||||||
from invokeai.app.services.session_queue.session_queue_common import (
|
from invokeai.app.services.session_queue.session_queue_common import (
|
||||||
QUEUE_ITEM_STATUS,
|
QUEUE_ITEM_STATUS,
|
||||||
@ -19,8 +20,6 @@ from invokeai.app.services.session_queue.session_queue_common import (
|
|||||||
)
|
)
|
||||||
from invokeai.app.services.shared.pagination import CursorPaginatedResults
|
from invokeai.app.services.shared.pagination import CursorPaginatedResults
|
||||||
|
|
||||||
from ..dependencies import ApiDependencies
|
|
||||||
|
|
||||||
session_queue_router = APIRouter(prefix="/v1/queue", tags=["queue"])
|
session_queue_router = APIRouter(prefix="/v1/queue", tags=["queue"])
|
||||||
|
|
||||||
|
|
||||||
|
@ -24,8 +24,8 @@ from invokeai.app.api.no_cache_staticfiles import NoCacheStaticFiles
|
|||||||
from invokeai.app.services.config.config_default import get_config
|
from invokeai.app.services.config.config_default import get_config
|
||||||
from invokeai.app.util.custom_openapi import get_openapi_func
|
from invokeai.app.util.custom_openapi import get_openapi_func
|
||||||
from invokeai.backend.util.devices import TorchDevice
|
from invokeai.backend.util.devices import TorchDevice
|
||||||
|
from invokeai.backend.util.logging import InvokeAILogger
|
||||||
|
|
||||||
from ..backend.util.logging import InvokeAILogger
|
|
||||||
from .api.dependencies import ApiDependencies
|
from .api.dependencies import ApiDependencies
|
||||||
from .api.routers import (
|
from .api.routers import (
|
||||||
app_info,
|
app_info,
|
||||||
|
@ -40,7 +40,7 @@ from invokeai.app.util.misc import uuid_string
|
|||||||
from invokeai.backend.util.logging import InvokeAILogger
|
from invokeai.backend.util.logging import InvokeAILogger
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from ..services.invocation_services import InvocationServices
|
from invokeai.app.services.invocation_services import InvocationServices
|
||||||
|
|
||||||
logger = InvokeAILogger.get_logger()
|
logger = InvokeAILogger.get_logger()
|
||||||
|
|
||||||
|
@ -14,8 +14,7 @@ from invokeai.app.invocations.fields import (
|
|||||||
from invokeai.app.invocations.model import ModelIdentifierField
|
from invokeai.app.invocations.model import ModelIdentifierField
|
||||||
from invokeai.app.services.shared.invocation_context import InvocationContext
|
from invokeai.app.services.shared.invocation_context import InvocationContext
|
||||||
from invokeai.app.util.controlnet_utils import CONTROLNET_MODE_VALUES, CONTROLNET_RESIZE_VALUES
|
from invokeai.app.util.controlnet_utils import CONTROLNET_MODE_VALUES, CONTROLNET_RESIZE_VALUES
|
||||||
|
from invokeai.version.invokeai_version import __version__
|
||||||
from ...version import __version__
|
|
||||||
|
|
||||||
|
|
||||||
class MetadataItemField(BaseModel):
|
class MetadataItemField(BaseModel):
|
||||||
|
@ -8,8 +8,8 @@ from invokeai.app.invocations.constants import LATENT_SCALE_FACTOR
|
|||||||
from invokeai.app.invocations.fields import FieldDescriptions, InputField, LatentsField, OutputField
|
from invokeai.app.invocations.fields import FieldDescriptions, InputField, LatentsField, OutputField
|
||||||
from invokeai.app.services.shared.invocation_context import InvocationContext
|
from invokeai.app.services.shared.invocation_context import InvocationContext
|
||||||
from invokeai.app.util.misc import SEED_MAX
|
from invokeai.app.util.misc import SEED_MAX
|
||||||
|
from invokeai.backend.util.devices import TorchDevice
|
||||||
|
|
||||||
from ...backend.util.devices import TorchDevice
|
|
||||||
from .baseinvocation import (
|
from .baseinvocation import (
|
||||||
BaseInvocation,
|
BaseInvocation,
|
||||||
BaseInvocationOutput,
|
BaseInvocationOutput,
|
||||||
|
@ -2,7 +2,7 @@ from typing import Optional
|
|||||||
|
|
||||||
from pydantic import Field
|
from pydantic import Field
|
||||||
|
|
||||||
from ..board_records.board_records_common import BoardRecord
|
from invokeai.app.services.board_records.board_records_common import BoardRecord
|
||||||
|
|
||||||
|
|
||||||
class BoardDTO(BoardRecord):
|
class BoardDTO(BoardRecord):
|
||||||
|
@ -3,16 +3,12 @@ from typing import Optional
|
|||||||
from PIL.Image import Image as PILImageType
|
from PIL.Image import Image as PILImageType
|
||||||
|
|
||||||
from invokeai.app.invocations.fields import MetadataField
|
from invokeai.app.invocations.fields import MetadataField
|
||||||
from invokeai.app.services.invoker import Invoker
|
from invokeai.app.services.image_files.image_files_common import (
|
||||||
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
|
|
||||||
from invokeai.app.services.shared.sqlite.sqlite_common import SQLiteDirection
|
|
||||||
|
|
||||||
from ..image_files.image_files_common import (
|
|
||||||
ImageFileDeleteException,
|
ImageFileDeleteException,
|
||||||
ImageFileNotFoundException,
|
ImageFileNotFoundException,
|
||||||
ImageFileSaveException,
|
ImageFileSaveException,
|
||||||
)
|
)
|
||||||
from ..image_records.image_records_common import (
|
from invokeai.app.services.image_records.image_records_common import (
|
||||||
ImageCategory,
|
ImageCategory,
|
||||||
ImageRecord,
|
ImageRecord,
|
||||||
ImageRecordChanges,
|
ImageRecordChanges,
|
||||||
@ -23,6 +19,10 @@ from ..image_records.image_records_common import (
|
|||||||
InvalidOriginException,
|
InvalidOriginException,
|
||||||
ResourceOrigin,
|
ResourceOrigin,
|
||||||
)
|
)
|
||||||
|
from invokeai.app.services.invoker import Invoker
|
||||||
|
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
|
||||||
|
from invokeai.app.services.shared.sqlite.sqlite_common import SQLiteDirection
|
||||||
|
|
||||||
from .images_base import ImageServiceABC
|
from .images_base import ImageServiceABC
|
||||||
from .images_common import ImageDTO, image_record_to_dto
|
from .images_common import ImageDTO, image_record_to_dto
|
||||||
|
|
||||||
|
@ -5,14 +5,13 @@ from abc import ABC, abstractmethod
|
|||||||
import torch
|
import torch
|
||||||
from typing_extensions import Self
|
from typing_extensions import Self
|
||||||
|
|
||||||
|
from invokeai.app.services.config.config_default import InvokeAIAppConfig
|
||||||
|
from invokeai.app.services.download.download_base import DownloadQueueServiceBase
|
||||||
|
from invokeai.app.services.events.events_base import EventServiceBase
|
||||||
from invokeai.app.services.invoker import Invoker
|
from invokeai.app.services.invoker import Invoker
|
||||||
|
from invokeai.app.services.model_install.model_install_base import ModelInstallServiceBase
|
||||||
from ..config import InvokeAIAppConfig
|
from invokeai.app.services.model_load.model_load_base import ModelLoadServiceBase
|
||||||
from ..download import DownloadQueueServiceBase
|
from invokeai.app.services.model_records.model_records_base import ModelRecordServiceBase
|
||||||
from ..events.events_base import EventServiceBase
|
|
||||||
from ..model_install import ModelInstallServiceBase
|
|
||||||
from ..model_load import ModelLoadServiceBase
|
|
||||||
from ..model_records import ModelRecordServiceBase
|
|
||||||
|
|
||||||
|
|
||||||
class ModelManagerServiceBase(ABC):
|
class ModelManagerServiceBase(ABC):
|
||||||
|
@ -6,17 +6,19 @@ from typing import Optional
|
|||||||
import torch
|
import torch
|
||||||
from typing_extensions import Self
|
from typing_extensions import Self
|
||||||
|
|
||||||
|
from invokeai.app.services.config.config_default import InvokeAIAppConfig
|
||||||
|
from invokeai.app.services.download.download_base import DownloadQueueServiceBase
|
||||||
|
from invokeai.app.services.events.events_base import EventServiceBase
|
||||||
from invokeai.app.services.invoker import Invoker
|
from invokeai.app.services.invoker import Invoker
|
||||||
|
from invokeai.app.services.model_install.model_install_base import ModelInstallServiceBase
|
||||||
|
from invokeai.app.services.model_install.model_install_default import ModelInstallService
|
||||||
|
from invokeai.app.services.model_load.model_load_base import ModelLoadServiceBase
|
||||||
|
from invokeai.app.services.model_load.model_load_default import ModelLoadService
|
||||||
|
from invokeai.app.services.model_records.model_records_base import ModelRecordServiceBase
|
||||||
from invokeai.backend.model_manager.load import ModelCache, ModelLoaderRegistry
|
from invokeai.backend.model_manager.load import ModelCache, ModelLoaderRegistry
|
||||||
from invokeai.backend.util.devices import TorchDevice
|
from invokeai.backend.util.devices import TorchDevice
|
||||||
from invokeai.backend.util.logging import InvokeAILogger
|
from invokeai.backend.util.logging import InvokeAILogger
|
||||||
|
|
||||||
from ..config import InvokeAIAppConfig
|
|
||||||
from ..download import DownloadQueueServiceBase
|
|
||||||
from ..events.events_base import EventServiceBase
|
|
||||||
from ..model_install import ModelInstallService, ModelInstallServiceBase
|
|
||||||
from ..model_load import ModelLoadService, ModelLoadServiceBase
|
|
||||||
from ..model_records import ModelRecordServiceBase
|
|
||||||
from .model_manager_base import ModelManagerServiceBase
|
from .model_manager_base import ModelManagerServiceBase
|
||||||
|
|
||||||
|
|
||||||
|
@ -46,6 +46,7 @@ from pathlib import Path
|
|||||||
from typing import List, Optional, Union
|
from typing import List, Optional, Union
|
||||||
|
|
||||||
from invokeai.app.services.shared.pagination import PaginatedResults
|
from invokeai.app.services.shared.pagination import PaginatedResults
|
||||||
|
from invokeai.app.services.shared.sqlite.sqlite_database import SqliteDatabase
|
||||||
from invokeai.backend.model_manager.config import (
|
from invokeai.backend.model_manager.config import (
|
||||||
AnyModelConfig,
|
AnyModelConfig,
|
||||||
BaseModelType,
|
BaseModelType,
|
||||||
@ -54,7 +55,6 @@ from invokeai.backend.model_manager.config import (
|
|||||||
ModelType,
|
ModelType,
|
||||||
)
|
)
|
||||||
|
|
||||||
from ..shared.sqlite.sqlite_database import SqliteDatabase
|
|
||||||
from .model_records_base import (
|
from .model_records_base import (
|
||||||
DuplicateModelException,
|
DuplicateModelException,
|
||||||
ModelRecordChanges,
|
ModelRecordChanges,
|
||||||
|
@ -13,6 +13,7 @@ from invokeai.app.services.events.events_common import (
|
|||||||
register_events,
|
register_events,
|
||||||
)
|
)
|
||||||
from invokeai.app.services.invocation_stats.invocation_stats_common import GESStatsNotFoundError
|
from invokeai.app.services.invocation_stats.invocation_stats_common import GESStatsNotFoundError
|
||||||
|
from invokeai.app.services.invoker import Invoker
|
||||||
from invokeai.app.services.session_processor.session_processor_base import (
|
from invokeai.app.services.session_processor.session_processor_base import (
|
||||||
OnAfterRunNode,
|
OnAfterRunNode,
|
||||||
OnAfterRunSession,
|
OnAfterRunSession,
|
||||||
@ -27,7 +28,6 @@ from invokeai.app.services.shared.graph import NodeInputError
|
|||||||
from invokeai.app.services.shared.invocation_context import InvocationContextData, build_invocation_context
|
from invokeai.app.services.shared.invocation_context import InvocationContextData, build_invocation_context
|
||||||
from invokeai.app.util.profiler import Profiler
|
from invokeai.app.util.profiler import Profiler
|
||||||
|
|
||||||
from ..invoker import Invoker
|
|
||||||
from .session_processor_base import InvocationServices, SessionProcessorBase, SessionRunnerBase
|
from .session_processor_base import InvocationServices, SessionProcessorBase, SessionRunnerBase
|
||||||
from .session_processor_common import SessionProcessorStatus
|
from .session_processor_common import SessionProcessorStatus
|
||||||
|
|
||||||
|
@ -5,9 +5,8 @@ from PIL import Image
|
|||||||
|
|
||||||
from invokeai.app.services.session_processor.session_processor_common import CanceledException, ProgressImage
|
from invokeai.app.services.session_processor.session_processor_common import CanceledException, ProgressImage
|
||||||
from invokeai.backend.model_manager.config import BaseModelType
|
from invokeai.backend.model_manager.config import BaseModelType
|
||||||
|
from invokeai.backend.stable_diffusion.diffusers_pipeline import PipelineIntermediateState
|
||||||
from ...backend.stable_diffusion import PipelineIntermediateState
|
from invokeai.backend.util.util import image_to_dataURL
|
||||||
from ...backend.util.util import image_to_dataURL
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from invokeai.app.services.events.events_base import EventServiceBase
|
from invokeai.app.services.events.events_base import EventServiceBase
|
||||||
|
@ -11,8 +11,8 @@ from PIL import Image
|
|||||||
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
|
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
|
||||||
|
|
||||||
from invokeai.backend.ip_adapter.ip_attention_weights import IPAttentionWeights
|
from invokeai.backend.ip_adapter.ip_attention_weights import IPAttentionWeights
|
||||||
|
from invokeai.backend.raw_model import RawModel
|
||||||
|
|
||||||
from ..raw_model import RawModel
|
|
||||||
from .resampler import Resampler
|
from .resampler import Resampler
|
||||||
|
|
||||||
|
|
||||||
|
@ -33,8 +33,7 @@ from typing_extensions import Annotated, Any, Dict
|
|||||||
from invokeai.app.invocations.constants import SCHEDULER_NAME_VALUES
|
from invokeai.app.invocations.constants import SCHEDULER_NAME_VALUES
|
||||||
from invokeai.app.util.misc import uuid_string
|
from invokeai.app.util.misc import uuid_string
|
||||||
from invokeai.backend.model_hash.hash_validator import validate_hash
|
from invokeai.backend.model_hash.hash_validator import validate_hash
|
||||||
|
from invokeai.backend.raw_model import RawModel
|
||||||
from ..raw_model import RawModel
|
|
||||||
|
|
||||||
# ModelMixin is the base class for all diffusers and transformers models
|
# ModelMixin is the base class for all diffusers and transformers models
|
||||||
# RawModel is the InvokeAI wrapper class for ip_adapters, loras, textual_inversion and onnx runtime
|
# RawModel is the InvokeAI wrapper class for ip_adapters, loras, textual_inversion and onnx runtime
|
||||||
|
@ -5,7 +5,7 @@ import psutil
|
|||||||
import torch
|
import torch
|
||||||
from typing_extensions import Self
|
from typing_extensions import Self
|
||||||
|
|
||||||
from ..util.libc_util import LibcUtil, Struct_mallinfo2
|
from invokeai.backend.model_manager.util.libc_util import LibcUtil, Struct_mallinfo2
|
||||||
|
|
||||||
GB = 2**30 # 1 GB
|
GB = 2**30 # 1 GB
|
||||||
|
|
||||||
|
@ -18,7 +18,7 @@ Use like this:
|
|||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
from typing import Callable, Dict, Optional, Tuple, Type, TypeVar
|
from typing import Callable, Dict, Optional, Tuple, Type, TypeVar
|
||||||
|
|
||||||
from ..config import (
|
from invokeai.backend.model_manager.config import (
|
||||||
AnyModelConfig,
|
AnyModelConfig,
|
||||||
BaseModelType,
|
BaseModelType,
|
||||||
ModelConfigBase,
|
ModelConfigBase,
|
||||||
@ -26,6 +26,7 @@ from ..config import (
|
|||||||
ModelType,
|
ModelType,
|
||||||
SubModelType,
|
SubModelType,
|
||||||
)
|
)
|
||||||
|
|
||||||
from . import ModelLoaderBase
|
from . import ModelLoaderBase
|
||||||
|
|
||||||
|
|
||||||
|
@ -13,8 +13,8 @@ from invokeai.backend.model_manager import (
|
|||||||
ModelType,
|
ModelType,
|
||||||
)
|
)
|
||||||
from invokeai.backend.model_manager.config import ControlNetCheckpointConfig, SubModelType
|
from invokeai.backend.model_manager.config import ControlNetCheckpointConfig, SubModelType
|
||||||
|
from invokeai.backend.model_manager.load.model_loader_registry import ModelLoaderRegistry
|
||||||
|
|
||||||
from .. import ModelLoaderRegistry
|
|
||||||
from .generic_diffusers import GenericDiffusersLoader
|
from .generic_diffusers import GenericDiffusersLoader
|
||||||
|
|
||||||
|
|
||||||
|
@ -18,8 +18,8 @@ from invokeai.backend.model_manager import (
|
|||||||
SubModelType,
|
SubModelType,
|
||||||
)
|
)
|
||||||
from invokeai.backend.model_manager.config import DiffusersConfigBase
|
from invokeai.backend.model_manager.config import DiffusersConfigBase
|
||||||
|
from invokeai.backend.model_manager.load.load_default import ModelLoader
|
||||||
from .. import ModelLoader, ModelLoaderRegistry
|
from invokeai.backend.model_manager.load.model_loader_registry import ModelLoaderRegistry
|
||||||
|
|
||||||
|
|
||||||
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.CLIPVision, format=ModelFormat.Diffusers)
|
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.CLIPVision, format=ModelFormat.Diffusers)
|
||||||
|
@ -15,9 +15,9 @@ from invokeai.backend.model_manager import (
|
|||||||
ModelType,
|
ModelType,
|
||||||
SubModelType,
|
SubModelType,
|
||||||
)
|
)
|
||||||
|
from invokeai.backend.model_manager.load.load_default import ModelLoader
|
||||||
from invokeai.backend.model_manager.load.model_cache.model_cache_base import ModelCacheBase
|
from invokeai.backend.model_manager.load.model_cache.model_cache_base import ModelCacheBase
|
||||||
|
from invokeai.backend.model_manager.load.model_loader_registry import ModelLoaderRegistry
|
||||||
from .. import ModelLoader, ModelLoaderRegistry
|
|
||||||
|
|
||||||
|
|
||||||
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.LoRA, format=ModelFormat.Diffusers)
|
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.LoRA, format=ModelFormat.Diffusers)
|
||||||
|
@ -13,8 +13,8 @@ from invokeai.backend.model_manager import (
|
|||||||
ModelType,
|
ModelType,
|
||||||
SubModelType,
|
SubModelType,
|
||||||
)
|
)
|
||||||
|
from invokeai.backend.model_manager.load.model_loader_registry import ModelLoaderRegistry
|
||||||
|
|
||||||
from .. import ModelLoaderRegistry
|
|
||||||
from .generic_diffusers import GenericDiffusersLoader
|
from .generic_diffusers import GenericDiffusersLoader
|
||||||
|
|
||||||
|
|
||||||
|
@ -25,9 +25,9 @@ from invokeai.backend.model_manager.config import (
|
|||||||
DiffusersConfigBase,
|
DiffusersConfigBase,
|
||||||
MainCheckpointConfig,
|
MainCheckpointConfig,
|
||||||
)
|
)
|
||||||
|
from invokeai.backend.model_manager.load.model_loader_registry import ModelLoaderRegistry
|
||||||
from invokeai.backend.util.silence_warnings import SilenceWarnings
|
from invokeai.backend.util.silence_warnings import SilenceWarnings
|
||||||
|
|
||||||
from .. import ModelLoaderRegistry
|
|
||||||
from .generic_diffusers import GenericDiffusersLoader
|
from .generic_diffusers import GenericDiffusersLoader
|
||||||
|
|
||||||
VARIANT_TO_IN_CHANNEL_MAP = {
|
VARIANT_TO_IN_CHANNEL_MAP = {
|
||||||
|
@ -12,10 +12,10 @@ from invokeai.backend.model_manager import (
|
|||||||
ModelType,
|
ModelType,
|
||||||
SubModelType,
|
SubModelType,
|
||||||
)
|
)
|
||||||
|
from invokeai.backend.model_manager.load.load_default import ModelLoader
|
||||||
|
from invokeai.backend.model_manager.load.model_loader_registry import ModelLoaderRegistry
|
||||||
from invokeai.backend.textual_inversion import TextualInversionModelRaw
|
from invokeai.backend.textual_inversion import TextualInversionModelRaw
|
||||||
|
|
||||||
from .. import ModelLoader, ModelLoaderRegistry
|
|
||||||
|
|
||||||
|
|
||||||
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.TextualInversion, format=ModelFormat.EmbeddingFile)
|
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.TextualInversion, format=ModelFormat.EmbeddingFile)
|
||||||
@ModelLoaderRegistry.register(
|
@ModelLoaderRegistry.register(
|
||||||
|
@ -12,8 +12,8 @@ from invokeai.backend.model_manager import (
|
|||||||
ModelType,
|
ModelType,
|
||||||
)
|
)
|
||||||
from invokeai.backend.model_manager.config import AnyModel, SubModelType, VAECheckpointConfig
|
from invokeai.backend.model_manager.config import AnyModel, SubModelType, VAECheckpointConfig
|
||||||
|
from invokeai.backend.model_manager.load.model_loader_registry import ModelLoaderRegistry
|
||||||
|
|
||||||
from .. import ModelLoaderRegistry
|
|
||||||
from .generic_diffusers import GenericDiffusersLoader
|
from .generic_diffusers import GenericDiffusersLoader
|
||||||
|
|
||||||
|
|
||||||
|
@ -18,8 +18,11 @@ from pydantic.networks import AnyHttpUrl
|
|||||||
from requests.sessions import Session
|
from requests.sessions import Session
|
||||||
|
|
||||||
from invokeai.backend.model_manager import ModelRepoVariant
|
from invokeai.backend.model_manager import ModelRepoVariant
|
||||||
|
from invokeai.backend.model_manager.metadata.metadata_base import (
|
||||||
from ..metadata_base import AnyModelRepoMetadata, AnyModelRepoMetadataValidator, BaseMetadata
|
AnyModelRepoMetadata,
|
||||||
|
AnyModelRepoMetadataValidator,
|
||||||
|
BaseMetadata,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class ModelMetadataFetchBase(ABC):
|
class ModelMetadataFetchBase(ABC):
|
||||||
|
@ -25,13 +25,13 @@ from pydantic.networks import AnyHttpUrl
|
|||||||
from requests.sessions import Session
|
from requests.sessions import Session
|
||||||
|
|
||||||
from invokeai.backend.model_manager.config import ModelRepoVariant
|
from invokeai.backend.model_manager.config import ModelRepoVariant
|
||||||
|
from invokeai.backend.model_manager.metadata.metadata_base import (
|
||||||
from ..metadata_base import (
|
|
||||||
AnyModelRepoMetadata,
|
AnyModelRepoMetadata,
|
||||||
HuggingFaceMetadata,
|
HuggingFaceMetadata,
|
||||||
RemoteModelFile,
|
RemoteModelFile,
|
||||||
UnknownMetadataException,
|
UnknownMetadataException,
|
||||||
)
|
)
|
||||||
|
|
||||||
from .fetch_base import ModelMetadataFetchBase
|
from .fetch_base import ModelMetadataFetchBase
|
||||||
|
|
||||||
HF_MODEL_RE = r"https?://huggingface.co/([\w\-.]+/[\w\-.]+)"
|
HF_MODEL_RE = r"https?://huggingface.co/([\w\-.]+/[\w\-.]+)"
|
||||||
|
@ -24,8 +24,7 @@ from requests.sessions import Session
|
|||||||
from typing_extensions import Annotated
|
from typing_extensions import Annotated
|
||||||
|
|
||||||
from invokeai.backend.model_manager import ModelRepoVariant
|
from invokeai.backend.model_manager import ModelRepoVariant
|
||||||
|
from invokeai.backend.model_manager.util.select_hf_files import filter_files
|
||||||
from ..util import select_hf_files
|
|
||||||
|
|
||||||
|
|
||||||
class UnknownMetadataException(Exception):
|
class UnknownMetadataException(Exception):
|
||||||
@ -112,9 +111,7 @@ class HuggingFaceMetadata(ModelMetadataWithFiles):
|
|||||||
session = session or Session()
|
session = session or Session()
|
||||||
configure_http_backend(backend_factory=lambda: session) # used in testing
|
configure_http_backend(backend_factory=lambda: session) # used in testing
|
||||||
|
|
||||||
paths = select_hf_files.filter_files(
|
paths = filter_files([x.path for x in self.files], variant, subfolder) # all files in the model
|
||||||
[x.path for x in self.files], variant, subfolder
|
|
||||||
) # all files in the model
|
|
||||||
prefix = f"{subfolder}/" if subfolder else ""
|
prefix = f"{subfolder}/" if subfolder else ""
|
||||||
# the next step reads model_index.json to determine which subdirectories belong
|
# the next step reads model_index.json to determine which subdirectories belong
|
||||||
# to the model
|
# to the model
|
||||||
|
@ -17,7 +17,7 @@ from dataclasses import dataclass
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Dict, List, Optional, Set
|
from typing import Dict, List, Optional, Set
|
||||||
|
|
||||||
from ..config import ModelRepoVariant
|
from invokeai.backend.model_manager.config import ModelRepoVariant
|
||||||
|
|
||||||
|
|
||||||
def filter_files(
|
def filter_files(
|
||||||
|
@ -10,7 +10,7 @@ import torch
|
|||||||
from onnx import numpy_helper
|
from onnx import numpy_helper
|
||||||
from onnxruntime import InferenceSession, SessionOptions, get_available_providers
|
from onnxruntime import InferenceSession, SessionOptions, get_available_providers
|
||||||
|
|
||||||
from ..raw_model import RawModel
|
from invokeai.backend.raw_model import RawModel
|
||||||
|
|
||||||
ONNX_WEIGHTS_NAME = "model.onnx"
|
ONNX_WEIGHTS_NAME = "model.onnx"
|
||||||
|
|
||||||
|
@ -206,7 +206,12 @@ ignore = [
|
|||||||
"B008", # https://docs.astral.sh/ruff/rules/function-call-in-default-argument/
|
"B008", # https://docs.astral.sh/ruff/rules/function-call-in-default-argument/
|
||||||
"B904", # https://docs.astral.sh/ruff/rules/raise-without-from-inside-except/
|
"B904", # https://docs.astral.sh/ruff/rules/raise-without-from-inside-except/
|
||||||
]
|
]
|
||||||
select = ["B", "C", "E", "F", "W", "I"]
|
select = ["B", "C", "E", "F", "W", "I", "TID"]
|
||||||
|
|
||||||
|
[tool.ruff.lint.flake8-tidy-imports]
|
||||||
|
# Disallow all relative imports.
|
||||||
|
ban-relative-imports = "parents"
|
||||||
|
|
||||||
#=== End: Ruff
|
#=== End: Ruff
|
||||||
|
|
||||||
#=== Begin: MyPy
|
#=== Begin: MyPy
|
||||||
|
Loading…
x
Reference in New Issue
Block a user