Use the new 'diffusionkit' package name

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

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()
 

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()
 

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()
 

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()