intuit / Trapheus

This tool automates restoration of RDS database instances from snapshots into any dev, staging or production environments. It supports individual RDS Snapshot as well as cluster snapshot restore operations.
https://intuit.github.io/Trapheus/
MIT License
106 stars 53 forks source link

Update requirements.txt to add accelerate and bitsandbytes #116

Closed stationeros closed 1 year ago

stationeros commented 1 year ago

Description

Adding https://github.com/TimDettmers/bitsandbytes Adding https://pypi.org/project/accelerate/

Adding the above to support GPU functions and load the model in 8bit

Screenshot 2023-09-16 at 10 07 38 PM

Thanks for contributing this Pull Request. Make sure that you submit this Pull Request against the master branch of this repository, add a brief description, and tag the relevant issue(s) and PR(s) below.