File tree 1 file changed +2
-2
lines changed
src/axolotl/integrations/llmcompressor_sft
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -13,7 +13,7 @@ class SFTArgs(BaseModel):
13
13
14
14
# Typing for recipe is set to Any due to:
15
15
# 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 ." )]
17
17
18
18
model_config = ConfigDict (
19
19
arbitrary_types_allowed = True ,
@@ -24,7 +24,7 @@ class SFTArgs(BaseModel):
24
24
class LLMCompressorArgs (BaseModel ):
25
25
"""LLMCompressor configuration BaseModel."""
26
26
27
- llmcompressor : Annotated [SFTArgs , Field (description = "SFT llmcompressor args " )]
27
+ llmcompressor : Annotated [SFTArgs , Field (description = "Arguments enabling compression pathways through the LLM Compressor plugins " )]
28
28
29
29
model_config = ConfigDict (
30
30
validate_assignment = True ,
You can’t perform that action at this time.
0 commit comments