return model in `load_model` function
b518069
Running
Dan Bochman
commited on
Spaces:
Running
on
Zero
Running
on
Zero
Commit History
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
initial commit without ckpt
cd4d90a
unverified
Dan Bochman
commited on