update params to supprot torch.jit.trace torchscript conversion 251565d katielink commited on Aug 16, 2023