Skip to content

GitLab

"examples/language/gpt/vscode:/vscode.git/clone" did not exist on "d83c633ca63c4eef49f3473aa998515fa5ca573f"
Unverified Commit 274c1a3b authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[fx] fixed apex normalization patch exception (#1352)

parent ce470ba3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/fx/tracer/meta_patch/patched_module/normalization.py
View file @ 274c1a3b
......@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment