Revert "[BugFix] Fix 'GPUARWorker' attribute error (#5478)"
This reverts commit f4640d9a2e.
This commit is contained in:
1 parent
c96f0bc773
commit
454b8a60be
1 file changed
-1
@@ -120,7 +120,6 @@ class OmniGPUWorkerBase(GPUWorker):
|
||||
self.model_runner.profile_run()
|
||||
|
||||
self.non_torch_memory = profile_result.non_torch_increase
|
||||
self.total_consumed = profile_result.total_consumed
|
||||
self.peak_activation_memory = profile_result.torch_peak_increase
|
||||
|
||||
process_memory = (
|
||||
|
||||
Reference in new issue
Block a user