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

Merge 34ca1f7d4b into sapling-pr-archive-EntilZha
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:
Pedro Rodriguez 2025-01-24 13:59:47 -08:00 committed by GitHub
parent f1a2589266 34ca1f7d4b
commit b1c12dd275
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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/train.py
View file

@ -47,6 +47,7 @@ from bytelatent.probe import AutoProbeD
from bytelatent.profiling import maybe_run_profiler from bytelatent.profiling import maybe_run_profiler
from bytelatent.stool import StoolArgs, launch_job from bytelatent.stool import StoolArgs, launch_job
from bytelatent.transformer import ( from bytelatent.transformer import (
LMTransformer,
build_fsdp_grouping_plan, build_fsdp_grouping_plan,
get_no_recompute_ops, get_no_recompute_ops,
get_num_flop_per_token, get_num_flop_per_token,