Update model loading logic and add support for multiple checkpoints 7576408 Dan Bochman commited on 25 days ago
cast input_tensor to cuda after acquiring zero gpu 6db27b1 unverified Dan Bochman commited on 25 days ago
Update model device to "cuda" after acquiring it with ZERO 3a1cfd0 unverified Dan Bochman commited on 25 days ago