Commit History

Update title in README.md
71556a8

Dan Bochman commited on

Update model loading logic and add support for multiple checkpoints
7576408

Dan Bochman commited on

attempt #31235 to work with ZERO
9a2f042
unverified

Dan Bochman commited on

move spaces.GPU to main segment scope
2233e64
unverified

Dan Bochman commited on

cast to cuda outside of spaces.GPU scope
a76f764
unverified

Dan Bochman commited on

cast input_tensor to cuda after acquiring zero gpu
6db27b1
unverified

Dan Bochman commited on

Update model device to "cuda" after acquiring it with ZERO
3a1cfd0
unverified

Dan Bochman commited on

remove casting to bf16
05346fa
unverified

Dan Bochman commited on

explictly cast to bfloat16
5310d0a
unverified

Dan Bochman commited on

adapt to use ZERO spaces.GPU
d46329a
unverified

Dan Bochman commited on

download checkpoint instead of uploading it
4375c79
unverified

Dan Bochman commited on

fixed README.md format
bf03d7d
unverified

Dan Bochman commited on

initial commit without ckpt
cd4d90a
unverified

Dan Bochman commited on