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

black and assert comment

Browse source
This commit is contained in:
Srini Iyer 2025-02-13 02:26:46 +00:00
parent 0ce2cd45ef
commit 00c7a6f194
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
bytelatent/train.py
View file

@ -130,7 +130,9 @@ def validate_train_args(args: TrainArgs, output_size: int):
if args.model is not None:
logger.info(f"Setting model output size to {args.model.vocab_size}")
args.model.vocab_size = output_size
assert(args.model.max_encoder_seq_length == args.data.max_encoder_seq_length)
assert (
args.model.max_encoder_seq_length == args.data.max_encoder_seq_length
), "max_encoder_seq_length for model and data should match"
if args.entropy_model is not None:
logger.info(f"Setting model output size to {args.entropy_model.vocab_size}")