diff --git a/service/utils.py b/service/aipg_utils.py similarity index 100% rename from service/utils.py rename to service/aipg_utils.py diff --git a/service/model_download_adpater.py b/service/model_download_adpater.py index 4ffa79d8..15430c8a 100644 --- a/service/model_download_adpater.py +++ b/service/model_download_adpater.py @@ -7,7 +7,7 @@ from psutil._common import bytes2human from model_downloader import HFPlaygroundDownloader import realesrgan -import utils +import aipg_utils as utils class Model_Downloader_Adapter: diff --git a/service/model_downloader.py b/service/model_downloader.py index 39a20f51..538da864 100644 --- a/service/model_downloader.py +++ b/service/model_downloader.py @@ -10,7 +10,7 @@ from exceptions import DownloadException import traceback import concurrent.futures -import utils +import aipg_utils as utils model_list_cache = dict() model_lock = Lock() diff --git a/service/paint_biz.py b/service/paint_biz.py index 29e853bd..485005eb 100644 --- a/service/paint_biz.py +++ b/service/paint_biz.py @@ -4,7 +4,7 @@ import random import time from typing import Any, Callable, Dict, List -import utils +import aipg_utils as utils import model_config import inpaint_utils from diffusers import ( diff --git a/service/rag.py b/service/rag.py index 10e54c09..29a2d00b 100644 --- a/service/rag.py +++ b/service/rag.py @@ -4,7 +4,7 @@ import time import os import re -import utils +import aipg_utils as utils import model_config from langchain_core.embeddings import Embeddings from langchain.text_splitter import RecursiveCharacterTextSplitter diff --git a/service/sd_adapter.py b/service/sd_adapter.py index 08f69bfa..2d58cb08 100644 --- a/service/sd_adapter.py +++ b/service/sd_adapter.py @@ -9,7 +9,7 @@ from psutil._common import bytes2human from PIL import Image import os -import utils +import aipg_utils as utils class SD_SSE_Adapter: diff --git a/service/web_api.py b/service/web_api.py index 03d08977..9055fd46 100644 --- a/service/web_api.py +++ b/service/web_api.py @@ -61,7 +61,7 @@ ) import paint_biz import llm_biz -import utils +import aipg_utils as utils import rag import model_config from model_downloader import HFPlaygroundDownloader