lines needed to include torch
Signed-off-by: scotthartbti <scott.hart.bti@gmail.com>

Change-Id: I81e92d26bf64b7695b7c8e20a7455dc1da855684