`core.safe_globals`#

Module Contents#

Functions#

register_safe_globals

Register megatron-core safe classes with torch serialization.

Data#

API#

core.safe_globals.SAFE_GLOBALS#: None

core.safe_globals.register_safe_globals()#: Register megatron-core safe classes with torch serialization.