[Bugfix] Fix normalize still being passed to PoolerConfig (#33794)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
@@ -174,8 +174,8 @@ class LLM:
|
||||
multi-modal processor obtained from `AutoProcessor.from_pretrained`.
|
||||
The available overrides depend on the model that is being run.
|
||||
For example, for Phi-3-Vision: `{"num_crops": 4}`.
|
||||
pooler_config: Initialize non-default pooling config for the pooling
|
||||
model. e.g. `PoolerConfig(seq_pooling_type="MEAN", normalize=False)`.
|
||||
pooler_config: Initialize non-default pooling config for the pooling model,
|
||||
e.g., `PoolerConfig(seq_pooling_type="MEAN", use_activation=False)`.
|
||||
compilation_config: Either an integer or a dictionary. If it is an
|
||||
integer, it is used as the mode of compilation optimization. If it
|
||||
is a dictionary, it can specify the full compilation configuration.
|
||||
|
||||
Reference in New Issue
Block a user