Merge branch 'master' into alvieirajr-swapping

microsoft · Apr 25, 2024 · fe8d400 · fe8d400
2 parents 71d4c84 + fcc731f
commit fe8d400
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepspeed/runtime/zero/partitioned_param_coordinator.py b/deepspeed/runtime/zero/partitioned_param_coordinator.py
@@ -34,6 +34,7 @@ def get_all_parameters(sub_module, recurse=False):
     return itertools.chain(sub_module.named_parameters(recurse=recurse), sub_module.ds_external_parameters())
 
 
+@compiler.disable
 def iter_params(module: Module, recurse=False) -> Iterable[Parameter]:
     return map(lambda pair: pair[1], get_all_parameters(module, recurse))