post merge cleanup

This commit is contained in:
Vladimir Mandic
2023-03-11 08:40:06 -05:00
parent b459794737
commit e0ea0151ed
2 changed files with 12 additions and 1 deletions
+11
View File
@@ -149,6 +149,7 @@ class Api:
self.add_api_route("/sdapi/v1/preprocess", self.preprocess, methods=["POST"], response_model=PreprocessResponse)
self.add_api_route("/sdapi/v1/train/embedding", self.train_embedding, methods=["POST"], response_model=TrainResponse)
self.add_api_route("/sdapi/v1/train/hypernetwork", self.train_hypernetwork, methods=["POST"], response_model=TrainResponse)
self.add_api_route("/sdapi/v1/shutdown", self.shutdown, methods=["POST"])
self.add_api_route("/sdapi/v1/memory", self.get_memory, methods=["GET"], response_model=MemoryResponse)
self.add_api_route("/sdapi/v1/scripts", self.get_scripts_list, methods=["GET"], response_model=ScriptsList)
@@ -523,6 +524,16 @@ class Api:
shared.state.end()
return TrainResponse(info="train embedding error: {error}".format(error=error))
def shutdown(self):
print('shutdown request received')
# from modules.shared import demo
# demo.close()
# time.sleep(0.5)
# import sys
# sys.exit(0)
import os
os._exit(0)
def get_memory(self):
try:
import os, psutil