minor fix

stanford-crfm · Nov 12, 2024 · c82302c · c82302c
1 parent 5408c75
commit c82302c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/levanter/main/sft.py b/src/levanter/main/sft.py
@@ -45,7 +45,7 @@ class DatasetType(str, Enum):
 
 
 @dataclass
-class SFTConfig):
+class SFTConfig:
     # inherit most of the config from TrainLmConfig
     trainer: TrainerConfig = field(default_factory=TrainerConfig)
     model: LmConfig = field(default_factory=LlamaConfig)