mirror of
https://github.com/facebookresearch/blt.git
synced 2025-02-22 13:02:14 +00:00
missed a print
This commit is contained in:
parent
8be8c85a63
commit
89deebc8f3
|
@ -28,6 +28,7 @@ try:
|
|||
|
||||
RMSNorm = FusedRMSNorm
|
||||
except (ImportError, ModuleNotFoundError):
|
||||
print("Apex not found. Using nn.RMSNorm")
|
||||
RMSNorm = nn.RMSNorm
|
||||
|
||||
|
||||
|
|
Loading…
Reference in a new issue