diff --git a/examples/quantization/brevitas/README.md b/examples/quantization/brevitas/README.md index 1f662459..eec3c35a 100644 --- a/examples/quantization/brevitas/README.md +++ b/examples/quantization/brevitas/README.md @@ -12,7 +12,7 @@ The example shows an example on how to quantize a decoder-class LLM model throug The examples were tested using: - `python>=3.9` (required for QOP ONNX export) -- `brevitas` installed from dev (`pip install git+https://github.com/Xilinx/brevitas.git@dev`) +- `brevitas>=0.11` - `torch>=2.2` - `transformers>=4.38.0` - `optimum>=1.17.0` @@ -22,7 +22,7 @@ The examples were tested using: Note, you can install all the prerequisites with: ```bash -pip install 'brevitas @ git+https://github.com/Xilinx/brevitas.git@dev' 'optimum-amd[brevitas] @ git+https://github.com/huggingface/optimum-amd.git@main' +pip install 'optimum-amd[brevitas] @ git+https://github.com/huggingface/optimum-amd.git@main' ``` ## Running the Example diff --git a/setup.py b/setup.py index 4eedceb7..9c716c7f 100644 --- a/setup.py +++ b/setup.py @@ -39,7 +39,7 @@ "ryzenai": ["opencv-python", "timm"], "zentorch": ["torch==2.2.1"], "brevitas": [ - "brevitas", + "brevitas>=0.11", "torch>=2.2", "datasets>=2.17", "onnx",