LAVIS icon indicating copy to clipboard operation
LAVIS copied to clipboard

Published 20 hours ago • verified publisher icon salesforce

torch.amp typo in blip2_models/blip2_t5.py

Open liuyang12 opened this issue 1 year ago • 0 comments

should be torch.autocast()

https://github.com/salesforce/LAVIS/blob/0e9a3ecfa5dc668761ce6c0e3cf827578e8593c4/lavis/models/blip2_models/blip2_t5.py#L218

liuyang12 avatar Feb 09 '23 23:02 liuyang12