mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-31 16:40:41 +00:00
[fx] fixed apex normalization patch exception (#1352)
This commit is contained in:
@@ -26,5 +26,5 @@ try:
|
||||
meta_patched_module.register(apex.normalization.FusedRMSNorm)(torch_nn_normalize)
|
||||
meta_patched_module.register(apex.normalization.MixedFusedLayerNorm)(torch_nn_normalize)
|
||||
meta_patched_module.register(apex.normalization.MixedFusedRMSNorm)(torch_nn_normalize)
|
||||
except ImportError:
|
||||
except (ImportError, AttributeError):
|
||||
pass
|
||||
|
Reference in New Issue
Block a user