GPT-K / model /gptk.py

Commit History

reduce GPU memory by keeping necessary modules of query_enc
d8c6a57

cwkuo commited on

save memory usage
40e6e04

cwkuo commited on

implement gpt-k demo
7962ed0

cwkuo commited on