[Feature] support clear data (#3601)

* [Feature] support clear data

* update

* fix

* fix

* fix

* fix

* fix

* fix

* fix
This commit is contained in:
ltd0924
2025-09-23 10:20:02 +08:00
committed by GitHub
parent 772f0156f3
commit 83720da79f
12 changed files with 85 additions and 1 deletions

View File

@@ -1923,6 +1923,10 @@ class GPUModelRunner(ModelRunnerBase):
self.dynamic_weight_manager._log_memory("dynamic weight manager clear all memory")
def clear_requests(self):
"""Dynamic model loader use to clear requests use for RL"""
self.share_inputs["stop_flags"][:] = True
def update_parameters(self, pid):
"""Dynamic model loader use to update parameters use for RL"""
# Update parameters