From a244a6873a6026d2157cda2298f196edbe2c8565 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Mon, 12 Dec 2022 20:46:11 +0530 Subject: [PATCH] Use the new 'diffusionkit' package name --- ui/sd_internal/__init__.py | 2 +- ui/sd_internal/model_manager.py | 4 ++-- ui/sd_internal/renderer.py | 4 ++-- ui/sd_internal/task_manager.py | 2 +- ui/server.py | 2 +- 5 files changed, 7 insertions(+), 7 deletions(-) diff --git a/ui/sd_internal/__init__.py b/ui/sd_internal/__init__.py index 7093562a..71073216 100644 --- a/ui/sd_internal/__init__.py +++ b/ui/sd_internal/__init__.py @@ -1,6 +1,6 @@ from pydantic import BaseModel -from modules.types import GenerateImageRequest +from diffusionkit.types import GenerateImageRequest class TaskData(BaseModel): request_id: str = None diff --git a/ui/sd_internal/model_manager.py b/ui/sd_internal/model_manager.py index f5e8a5c7..18418624 100644 --- a/ui/sd_internal/model_manager.py +++ b/ui/sd_internal/model_manager.py @@ -4,8 +4,8 @@ import picklescan.scanner import rich from sd_internal import app, TaskData -from modules import model_loader -from modules.types import Context +from diffusionkit import model_loader +from diffusionkit.types import Context log = logging.getLogger() diff --git a/ui/sd_internal/renderer.py b/ui/sd_internal/renderer.py index 11a38da8..1fff48e1 100644 --- a/ui/sd_internal/renderer.py +++ b/ui/sd_internal/renderer.py @@ -10,8 +10,8 @@ import logging from sd_internal import device_manager from sd_internal import TaskData, Response, Image as ResponseImage, UserInitiatedStop -from modules import model_loader, image_generator, image_utils, filters as image_filters, data_utils -from modules.types import Context, GenerateImageRequest, FilterImageRequest +from diffusionkit import model_loader, image_generator, image_utils, filters as image_filters, data_utils +from diffusionkit.types import Context, GenerateImageRequest, FilterImageRequest log = logging.getLogger() diff --git a/ui/sd_internal/task_manager.py b/ui/sd_internal/task_manager.py index ef2a4a3b..f4576372 100644 --- a/ui/sd_internal/task_manager.py +++ b/ui/sd_internal/task_manager.py @@ -15,7 +15,7 @@ import queue, threading, time, weakref from typing import Any, Hashable from sd_internal import TaskData, device_manager -from modules.types import GenerateImageRequest +from diffusionkit.types import GenerateImageRequest log = logging.getLogger() diff --git a/ui/server.py b/ui/server.py index 42bdb0f4..7308dfc8 100644 --- a/ui/server.py +++ b/ui/server.py @@ -15,7 +15,7 @@ from pydantic import BaseModel from sd_internal import app, model_manager, task_manager from sd_internal import TaskData -from modules.types import GenerateImageRequest +from diffusionkit.types import GenerateImageRequest log = logging.getLogger()