Commit History

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