Skip to content

Commit 1b88f57

Browse files
tianyu-lgithubsgi
authored andcommitted
refactor TrainSpec to remove the name field (pytorch#1850)
1 parent 1432c09 commit 1b88f57

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torchtitan/experiments/vlm/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from dataclasses import fields
88
from typing import Any
99

10+
from torchtitan.components.loss import build_cross_entropy_loss
1011
from torchtitan.components.lr_scheduler import build_lr_schedulers
1112
from torchtitan.components.optimizer import build_optimizers
1213
from torchtitan.components.tokenizer import build_hf_tokenizer

0 commit comments

Comments
 (0)