vikarti.anatra/blt
1
0
Fork 0
mirror of https://github.com/facebookresearch/blt.git synced 2025-02-23 05:22:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

missed a print
Some checks failed
Lint with Black / lint (push) Has been cancelled
Details
Lint with isort / lint (push) Has been cancelled
Details

Browse source
This commit is contained in:
Srini Iyer 2025-02-14 19:21:28 +00:00
parent 8be8c85a63
commit 89deebc8f3
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
bytelatent/transformer.py
View file

@ -28,6 +28,7 @@ try:
RMSNorm = FusedRMSNorm RMSNorm = FusedRMSNorm
except (ImportError, ModuleNotFoundError): except (ImportError, ModuleNotFoundError):
print("Apex not found. Using nn.RMSNorm")
RMSNorm = nn.RMSNorm RMSNorm = nn.RMSNorm