mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
add more job state updates and update history tab
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -160,6 +160,7 @@ class Processor():
|
||||
self.load_config[k] = v
|
||||
|
||||
def load(self, processor_id: str = None, force: bool = True) -> str:
|
||||
from modules.shared import state
|
||||
try:
|
||||
t0 = time.time()
|
||||
processor_id = processor_id or self.processor_id
|
||||
@@ -179,6 +180,7 @@ class Processor():
|
||||
cls = config[processor_id]['class']
|
||||
# log.debug(f'Control Processor loading: id="{processor_id}" class={cls.__name__}')
|
||||
debug(f'Control Processor config={self.load_config}')
|
||||
jobid = state.begin('Load processor')
|
||||
if 'DWPose' in processor_id:
|
||||
det_ckpt = 'https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_l_8x8_300e_coco/yolox_l_8x8_300e_coco_20211126_140236-d3bd2b23.pth'
|
||||
if 'Tiny' == config['DWPose']['model']:
|
||||
@@ -209,6 +211,7 @@ class Processor():
|
||||
else:
|
||||
self.model = cls() # class instance only
|
||||
t1 = time.time()
|
||||
state.end(jobid)
|
||||
self.processor_id = processor_id
|
||||
log.debug(f'Control Processor loaded: id="{processor_id}" class={self.model.__class__.__name__} time={t1-t0:.2f}')
|
||||
return f'Processor loaded: {processor_id}'
|
||||
|
||||
Reference in New Issue
Block a user