Allow running with non working

This commit is contained in:
pythongosssss 2025-03-28 11:46:05 +08:00
parent 7bf381bc9e
commit fde9fdddff
2 changed files with 10 additions and 3 deletions

View File

@ -111,6 +111,8 @@ def init_db():
session.commit()
def can_create_session():
return Session is not None
def create_session():
return Session()

11
main.py
View File

@ -138,7 +138,7 @@ import server
from server import BinaryEventTypes
import nodes
import comfy.model_management
from app.database.db import init_db
from app.database.db import can_create_session, init_db
from app.model_processor import model_processor
def cuda_malloc_warning():
@ -264,7 +264,11 @@ def start_comfyui(asyncio_loop=None):
cuda_malloc_warning()
init_db()
try:
init_db()
except Exception as e:
logging.error(f"Failed to initialize database. Please report this error as in future the database will be required: {e}")
prompt_server.add_routes()
hijack_progress(prompt_server)
@ -274,7 +278,8 @@ def start_comfyui(asyncio_loop=None):
exit(0)
# Scan for changed model files and update db
model_processor.run()
if can_create_session():
model_processor.run()
os.makedirs(folder_paths.get_temp_directory(), exist_ok=True)
call_on_start = None