holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use the new 'diffusionkit' package name

Browse Source
This commit is contained in:
cmdr2 2022-12-12 20:46:11 +05:30
parent ceff4f06c1
commit a244a6873a
5 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/sd_internal/__init__.py
View File

@ -1,6 +1,6 @@
from pydantic import BaseModel from pydantic import BaseModel
from modules.types import GenerateImageRequest from diffusionkit.types import GenerateImageRequest
class TaskData(BaseModel): class TaskData(BaseModel):
request_id: str = None request_id: str = None

4
ui/sd_internal/model_manager.py
View File

@ -4,8 +4,8 @@ import picklescan.scanner
import rich import rich
from sd_internal import app, TaskData from sd_internal import app, TaskData
from modules import model_loader from diffusionkit import model_loader
from modules.types import Context from diffusionkit.types import Context
log = logging.getLogger() log = logging.getLogger()

4
ui/sd_internal/renderer.py
View File

@ -10,8 +10,8 @@ import logging
from sd_internal import device_manager from sd_internal import device_manager
from sd_internal import TaskData, Response, Image as ResponseImage, UserInitiatedStop 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 diffusionkit import model_loader, image_generator, image_utils, filters as image_filters, data_utils
from modules.types import Context, GenerateImageRequest, FilterImageRequest from diffusionkit.types import Context, GenerateImageRequest, FilterImageRequest
log = logging.getLogger() log = logging.getLogger()

2
ui/sd_internal/task_manager.py
View File

@ -15,7 +15,7 @@ import queue, threading, time, weakref
from typing import Any, Hashable from typing import Any, Hashable
from sd_internal import TaskData, device_manager from sd_internal import TaskData, device_manager
from modules.types import GenerateImageRequest from diffusionkit.types import GenerateImageRequest
log = logging.getLogger() log = logging.getLogger()

2
ui/server.py
View File

@ -15,7 +15,7 @@ from pydantic import BaseModel
from sd_internal import app, model_manager, task_manager from sd_internal import app, model_manager, task_manager
from sd_internal import TaskData from sd_internal import TaskData
from modules.types import GenerateImageRequest from diffusionkit.types import GenerateImageRequest
log = logging.getLogger() log = logging.getLogger()