LAVIS icon indicating copy to clipboard operation
LAVIS copied to clipboard

Published 20 hours ago • verified publisher icon salesforce

Error when always using model in cpu

Open Jack-ZC8 opened this issue 1 year ago • 1 comments

https://github.com/salesforce/LAVIS/blob/3ac397aa075c3e60b9521b012dda3660e3e35f1e/lavis/runners/runner_base.py#L80

Hello, I get self.model is None when I set 'cpu' as my device. I find that if my device is 'cpu', it won't move the model to device and returned self._wrapped_model will be None.

Jack-ZC8 avatar Feb 15 '23 06:02 Jack-ZC8