Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

lora training

Browse Source
This commit is contained in:
Vladimir Mandic
2023-02-11 17:34:20 -05:00
parent 89502b9552
commit 148c4567d6
21 changed files with 248 additions and 181 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cli/modules/sdapi.py
+11 -2
View File
@@ -34,7 +34,7 @@ async def result(req):
res = {}
else:
res = Map(json)
log.debug({ 'request': req.status, 'url': req.url, 'reason': req.reason, 'result': res })
log.debug({ 'request': req.status, 'url': req.url, 'reason': req.reason })
return res
@@ -51,7 +51,7 @@ def resultsync(req: requests.Response):
res = {}
else:
res = Map(json)
log.debug({ 'request': req.status_code, 'url': req.url, 'reason': req.reason, 'result': res })
log.debug({ 'request': req.status_code, 'url': req.url, 'reason': req.reason })
return res
@@ -128,6 +128,13 @@ async def progress():
return res
def shutdown():
try:
postsync('/sdapi/v1/shutdown')
except Exception as e:
log.info({ 'shutdown': e })
async def session():
global sess # pylint: disable=global-statement
time = aiohttp.ClientTimeout(total = None, sock_connect = 10, sock_read = None) # default value is 5 minutes, we need longer for training
@@ -161,4 +168,6 @@ if __name__ == "__main__":
asyncio.run(interrupt())
if 'progress' in sys.argv:
asyncio.run(progress())
if 'shutdown' in sys.argv:
shutdown()
asyncio.run(close())