mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
Enable ^C during app startup when hashing models
This commit is contained in:
parent
ddf917f68c
commit
fc02da4627
@ -1,6 +1,9 @@
|
|||||||
import asyncio
|
import asyncio
|
||||||
import mimetypes
|
import mimetypes
|
||||||
|
import os
|
||||||
|
import signal
|
||||||
import socket
|
import socket
|
||||||
|
import sys
|
||||||
from contextlib import asynccontextmanager
|
from contextlib import asynccontextmanager
|
||||||
from inspect import signature
|
from inspect import signature
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
@ -223,6 +226,27 @@ app.mount(
|
|||||||
|
|
||||||
|
|
||||||
def invoke_api() -> None:
|
def invoke_api() -> None:
|
||||||
|
class InterruptWatcher:
|
||||||
|
def __init__(self):
|
||||||
|
self.child = os.fork()
|
||||||
|
if self.child == 0:
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
self.watch()
|
||||||
|
|
||||||
|
def watch(self) -> None:
|
||||||
|
try:
|
||||||
|
os.wait()
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
self.kill()
|
||||||
|
sys.exit()
|
||||||
|
|
||||||
|
def kill(self) -> None:
|
||||||
|
try:
|
||||||
|
os.kill(self.child, signal.SIGKILL)
|
||||||
|
except OSError:
|
||||||
|
pass
|
||||||
|
|
||||||
def find_port(port: int) -> int:
|
def find_port(port: int) -> int:
|
||||||
"""Find a port not in use starting at given port"""
|
"""Find a port not in use starting at given port"""
|
||||||
# Taken from https://waylonwalker.com/python-find-available-port/, thanks Waylon!
|
# Taken from https://waylonwalker.com/python-find-available-port/, thanks Waylon!
|
||||||
@ -233,6 +257,8 @@ def invoke_api() -> None:
|
|||||||
else:
|
else:
|
||||||
return port
|
return port
|
||||||
|
|
||||||
|
InterruptWatcher()
|
||||||
|
|
||||||
if app_config.dev_reload:
|
if app_config.dev_reload:
|
||||||
try:
|
try:
|
||||||
import jurigged
|
import jurigged
|
||||||
|
Loading…
Reference in New Issue
Block a user