Skip to content

Commit 5cf596c

Browse files
rahul-tulimarkurtz
andauthored
Apply suggestions from @markurtz
Co-authored-by: Mark Kurtz <mark.j.kurtz@gmail.com>
1 parent 205e8ba commit 5cf596c

File tree

1 file changed

+2
-2
lines changed
  • src/axolotl/integrations/llmcompressor_sft

1 file changed

+2
-2
lines changed

src/axolotl/integrations/llmcompressor_sft/args.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ class SFTArgs(BaseModel):
1313

1414
# Typing for recipe is set to Any due to:
1515
# https://github.com/vllm-project/llm-compressor/issues/1319
16-
recipe: Annotated[Any, Field(description="Recipe config.")]
16+
recipe: Annotated[Any, Field(description="The recipe containing the compression algorithms and hyperparameters to apply.")]
1717

1818
model_config = ConfigDict(
1919
arbitrary_types_allowed=True,
@@ -24,7 +24,7 @@ class SFTArgs(BaseModel):
2424
class LLMCompressorArgs(BaseModel):
2525
"""LLMCompressor configuration BaseModel."""
2626

27-
llmcompressor: Annotated[SFTArgs, Field(description="SFT llmcompressor args")]
27+
llmcompressor: Annotated[SFTArgs, Field(description="Arguments enabling compression pathways through the LLM Compressor plugins")]
2828

2929
model_config = ConfigDict(
3030
validate_assignment=True,

0 commit comments

Comments
 (0)