Add torch
as a requirement
#22
Merged
Loading
torch
as a requirement
#22