Defer torch import to live-mode codepath for demo mode compatibility 3320861 WolfDavid commited on Apr 10