Rename utils.py to be compatible with ComfyUI in embeddable python env

TNG · Nov 21, 2024 · 98c55b2 · 98c55b2
1 parent af8c489
commit 98c55b2
Show file tree

Hide file tree

Showing 7 changed files with 6 additions and 6 deletions.
diff --git a/service/utils.py → service/aipg_utils.py b/service/utils.py → service/aipg_utils.py
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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