This commit is contained in:
Dan Saunders
2025-08-30 00:04:39 -04:00
parent 02f308351c
commit 98f230d864
3 changed files with 3 additions and 6 deletions

View File

@@ -31,10 +31,7 @@ PRs are **greatly welcome**!
Please run below to setup env
```bash
# Using uv (recommended)
uv pip install -e .[dev]
# OR using pip
pip3 install -e .[dev]
pre-commit install

View File

@@ -117,7 +117,7 @@ deepspeed = [
]
mamba-ssm = [
"mamba-ssm>=2.2.0", # Updated to latest stable version
"mamba-ssm>=2.2.0",
"causal_conv1d>=1.4.0",
]

View File

@@ -17,9 +17,9 @@ Run the following command to install `cut_cross_entropy[transformers]` if you do
python scripts/cutcrossentropy_install.py | sh
```
- If you are installing from pip
- If you are installing manually
```bash
pip3 uninstall -y cut-cross-entropy && pip3 install "cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@147ea28"
uv pip uninstall -y cut-cross-entropy && uv pip install "cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@c6a32c5"
```
## Usage