File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
src/llmcompressor/entrypoints Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -241,7 +241,7 @@ def oneshot(
241241 dataset : str | "Dataset" | "DatasetDict" | None = None ,
242242 dataset_config_name : str | None = None ,
243243 dataset_path : str | None = None ,
244- splits : str | list | dict | None = None ,
244+ splits : str | list [ str ] | dict [ str , str ] | None = None ,
245245 num_calibration_samples : int = 512 ,
246246 shuffle_calibration_samples : bool = True ,
247247 max_seq_length : int = 384 ,
@@ -287,7 +287,8 @@ def oneshot(
287287 tag, or commit id).
288288
289289 # Recipe arguments
290- :param recipe: Path to a LLM Compressor sparsification recipe.
290+ :param recipe: Path to a LLM Compressor sparsification recipe, or a list of paths
291+ to multiple recipes.
291292 :param recipe_args: List of recipe arguments to evaluate, in the
292293 format "key1=value1", "key2=value2".
293294 :param clear_sparse_session: Whether to clear CompressionSession/
You can’t perform that action at this time.
0 commit comments