mirror of
https://github.com/wassname/activation_store.git
synced 2026-06-27 16:44:07 +08:00
doc: use keyword arguments in the example
Signed-off-by: thiswillbeyourgithub <26625900+thiswillbeyourgithub@users.noreply.github.com>
This commit is contained in:
@@ -30,7 +30,7 @@ layer_groups = {'mlp.down_proj': [
|
||||
'model.layers.23.mlp.up_proj']}
|
||||
|
||||
# collect activations into a huggingface dataset
|
||||
f = activation_store(ds, model, layers=layer_groups)
|
||||
f = activation_store(loader=ds, model=model, layers=layer_groups)
|
||||
f
|
||||
# > Generating train split: 0 examples [00:00, ? examples/s]
|
||||
# Dataset({
|
||||
|
||||
Reference in New Issue
Block a user