-
Notifications
You must be signed in to change notification settings - Fork 5
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Integration of CTransformers for benchmarks. #70
Conversation
For, now only llama is supported, since it has all the three types of device support, and also it is a standard model.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@Anindyadeep I left some comments :) furthermore, you need to update also the README.md.template with the values for ctranslate on cpu, metal (M2 Pro) and cuda on A100
bench_ctransformers/requirements.txt
Outdated
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pin at least the major versions
Co-authored-by: Nicola Sosio <[email protected]>
Co-authored-by: Nicola Sosio <[email protected]>
merge from main.
merge from main.
- integrated setup.sh file installations. - removed un-necessary package installation inside bench.sh
bench_ctransformers/setup.sh
Outdated
source "$VENV_DIR/bin/activate" | ||
fi | ||
|
||
install_device_specific_ctransformers "$DEVICE" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
can't we move this after installing the requirements (line 79) to avoid reinstalling it every time? and remove the activation of the environment?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ahh yes
Co-authored-by: Nicola Sosio <[email protected]>
- reverted back the conditional statement. - placed install_ctransformers_cuda() function after requirements installation.
This PR integrates CTransformers to allow user to benchmark models used using CTransformers.
Addresses Issue #22