Skip to content

Commit 7a8043a

Browse files
committed
fix: make linter happy
Signed-off-by: Oleg S <[email protected]>
1 parent d025471 commit 7a8043a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/instructlab/training/config.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ class DeepSpeedOffloadStrategy(Enum):
3030

3131
# public API
3232
class DistributedBackend(Enum):
33-
FSDP: str = "fsdp"
34-
DEEPSPEED: str = "deepspeed"
33+
FSDP = "fsdp"
34+
DEEPSPEED = "deepspeed"
3535

3636

3737
# public API
@@ -121,6 +121,7 @@ class DeepSpeedOptions(BaseModel):
121121
save_samples: int | None = None
122122

123123

124+
# public API
124125
class DistillationConfig(BaseModel):
125126
"""
126127
Config to use when performing knowledge distillation during training.

0 commit comments

Comments
 (0)