diff --git a/env/activate b/env/activate index 5c06b577..ebca09fc 100644 --- a/env/activate +++ b/env/activate @@ -20,7 +20,7 @@ else python3.11 -m venv $TTMLIR_VENV_DIR source $TTMLIR_VENV_DIR/bin/activate pip install --upgrade pip - pip install -r requirements.txt + python3.11 -m pip install -r requirements.txt fi export TTTORCH_ENV_ACTIVATED=1 export TTMLIR_ENV_ACTIVATED=1 diff --git a/requirements.txt b/requirements.txt index fecee534..4ee83327 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,10 @@ torch@https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.5.0%2Bcpu.cxx11.abi torchvision --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://github.com/llvm/torch-mlir-release/releases/expanded_assets/dev-wheels +black +ninja +pre-commit pybind11 -tabulate pytest +tabulate transformers