fix: change torch to lightblub