FIX AutoDL调度修复暖池功能

This commit is contained in:
2025-04-17 12:00:50 +08:00
parent 1272a33718
commit bf4c66fc27
3 changed files with 27 additions and 12 deletions

View File

@@ -53,20 +53,17 @@ class Server:
return {"status": "queuing", "msg":""}
def introspect_instance(self):
loguru.logger.info("Introspecting worker started")
loguru.logger.info("Introspecting worker started || Scaledown Window: %ds" % self.instance_pool.scaledown_window)
while True:
self.instance_pool.introspection()
time.sleep(0.5)
time.sleep(1)
def scaling_worker(self):
loguru.logger.info("Scaling worker started")
try:
last_target = 0
while True:
# 提交任务
if last_target != self.waiting_queue.get_size()+self.running_pool.get_running_size():
last_target = self.waiting_queue.get_size()+self.running_pool.get_running_size()
self.instance_pool.scale_instance(last_target)
self.instance_pool.scale_instance(self.waiting_queue.get_size()+self.running_pool.get_running_size(), disable_shrink=True)
for instance in self.instance_pool.instances:
if not instance.active:
# 从等待队列取出任务