The goal of this project is to generate activations from a Torch model. These activations will then be usable in ZML.
- Python 3.11
- Clone this repository to your local machine:
git clone https://github.com/vctrmn/zml-torch-activation-example.git
- Create a python virtual env
python3 -m venv venv
- Activate the virtual env
. venv/bin/activate
- Install the dependencies
pip install -r requirements.txt
- Activate the virtual env
. venv/bin/activate
- Run the activation script you want
python -m src.modern_bert
This project is licensed under the MIT License. See the LICENSE file for details.