Commit History

tune some default params
6b2ffd3

cwkuo commited on

disable beam search as it may cause OoM
ef2dc13

cwkuo commited on

reduce GPU memory by keeping necessary modules of query_enc
d8c6a57

cwkuo commited on

some quick fix
6855619

cwkuo commited on

save memory usage
40e6e04

cwkuo commited on

some quick fix
bf5fb05

cwkuo commited on

add requirements.txt
9d121b0

cwkuo commited on

code clean up
9051af7

cwkuo commited on

some quick fix
507ba2d

cwkuo commited on

update gitignore
9d53fd1

cwkuo commited on

trim checkpoint model weights
febd802

cwkuo commited on

implement gpt-k demo
7962ed0

cwkuo commited on

initial commit
fb92e97

cwkuo commited on