Skip to content

Commit

Permalink
dropped left over arg doc
Browse files Browse the repository at this point in the history
  • Loading branch information
tGhattas committed Jul 31, 2024
1 parent 7298e08 commit 2ecc994
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/transformers/models/mamba/configuration_mamba.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,8 +81,6 @@ class MambaConfig(PretrainedConfig):
Whether or not the cache should be used.
use_mambapy (`bool`, *optional*, defaults to `False`):
Determines the fallback strategy during training if the CUDA-based official implementation of Mamba is not avaiable. If `True`, the mamba.py implementation is used. If `False`, the naive and slower implementation is used. Consider switching to the naive version if memory is limited.
ignore_keys_for_inference (`List[str]`, *optional*, defaults to `[]`):
Example:
```python
Expand Down

0 comments on commit 2ecc994

Please sign in to comment.