mixtral-offloading
mixtral-offloading copied to clipboard
CLI interface added
This PR adds a small CLI interface to the repository which makes local usage easy.
Hey, @NJannasch! Thanks for the PR and sorry for the long reply. We would like to make sure the script works before merging it. Could you maybe make a Colab notebook that runs the script?
Hi @dvmazur, You can just checkout the branch, install dependencies and then call python3 cli.py as described in my README update.
An example Colab is available here (with some interestingly long responses lol): https://colab.research.google.com/drive/1Dkh0WNJAZD73Q4l4R0Y0T699mnAn6M5f?usp=sharing
I tried it out. This script works. Please merge if possible.