Skip to content

Commit

Permalink
Update src/sparseml/modifiers/obcq/base.py
Browse files Browse the repository at this point in the history
Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com>
  • Loading branch information
Satrat and bfineran committed Oct 9, 2023
1 parent 0a0cff9 commit 5fa7361
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/sparseml/modifiers/obcq/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ class SparseGPTModifier(Modifier):
True saves on GPU memory
:param prunen: N for N:M pruning
:param prunem: M for N:M pruning
:param compress_layers: list of layer names to compress during OBCQ, or ALL_TOKEN
:param compress_layers: list of layer names to compress during OBCQ, or '__ALL__'
to compress every layer in the model
:param target_ids: list of keys in model output to cache
:param layer_prefix: name of model attribute that contains the list of layers, i.e.
Expand Down

0 comments on commit 5fa7361

Please sign in to comment.