langchain icon indicating copy to clipboard operation
langchain copied to clipboard

WIP Accelerate enabled open source models

Open thepok opened this issue 2 years ago • 2 comments

Many TODOs but already workable.

Now its easy to load big open source models.

I was able to load neox 20b and facebook 30b on my laptop with 64gb ram and 16gb vram.

thepok avatar Dec 23 '22 21:12 thepok

nice! im not super familiar with accerlated stuff. how generic is the get_accelerated_gpt code?

@hwchase17 Since I ran into similar issue and I am familiar with this, I left a bunch of comments to @thepok in a review.

@thepok Besides the review comments, can you add an example of how Accelarate is used (preferably not as a notebook for review purposes)? Also Accelerate is very general purpose name to use for this. In general, try leaving generic names for use in langchain core.

delip avatar Jan 22 '23 22:01 delip

@delip Thank you for your time reviewing this. i have no interest at the moment to implement this further. Feel free to implement your suggestions your self. It could be possible to merge it in the huggingface llm loader for maximum impact.

thepok avatar Jan 23 '23 07:01 thepok

stale

dev2049 avatar May 15 '23 21:05 dev2049

ugh, so this have not been integrated?

Bec-k avatar Sep 07 '23 10:09 Bec-k

nope, but i think huggingfaces transformer lib can it now by itself, and hugginfaces transformer is already in langchain :)

thepok avatar Sep 07 '23 18:09 thepok