eaeea

Author: 3outeille

torchtitan/protocols/train_spec.py

@@ -77,7 +77,6 @@ def get_train_spec(name: str) -> TrainSpec:
     from torchtitan.experiments import _supported_experiments
     from torchtitan.models import _supported_models
 
-    # If name contains '/', treat it as a HuggingFace model ID and use transformers_backend
     if "/" in name:
         module = import_module("torchtitan.experiments.transformers_backend")
         return module.get_train_spec()