GPT-K / model

Commit History

reduce GPU memory by keeping necessary modules of query_enc
d8c6a57

cwkuo commited on

save memory usage
40e6e04

cwkuo commited on

some quick fix
bf5fb05

cwkuo commited on

code clean up
9051af7

cwkuo commited on

some quick fix
507ba2d

cwkuo commited on

trim checkpoint model weights
febd802

cwkuo commited on

implement gpt-k demo
7962ed0

cwkuo commited on