Spaces:
Runtime error
Runtime error
File size: 75 Bytes
ed66a8c |
1 2 3 4 |
from setup_device import get_device
device = get_device()
model.to(device) |