fix

AlibabaPAI · Nov 8, 2024 · 207237d · 207237d
1 parent 7b0cbc3
commit 207237d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/llumnix/llumlet/llumlet.py b/llumnix/llumlet/llumlet.py
@@ -140,6 +140,8 @@ async def migrate_out(self, dst_instance_name: str, num_requests: int) -> List[s
                 migrate_out_request = self.migration_scheduler.get_migrate_out_request()
                 if migrate_out_request is None:
                     break
+
+                migrate_out_request.migrating = True
                 logger.info("{}->{} begin migrate out {}".format(self.instance_id, dst_instance_id, migrate_out_request.request_id))
                 status = await self.migration_coordinator.migrate_out_multistage(migrate_in_ray_actor, migrate_out_request)