Lunderberg opened a new pull request, #16785: URL: https://github.com/apache/tvm/pull/16785
Prior to this commit, the weights used by `nn.Module` instances were required to be `nn.Parameter` instances. This commit allows the weights to instead be `nn.Tensor` instances, defined in terms of other `nn.Parameter` weights. This allows a model to define both the original weights that would be present in an external checkpoint (e.g. a Pytorch or Safetensors file), and the pre-processing that should be performed on those weights. This is a re-implementation of https://github.com/apache/tvm/pull/16757, which was reverted in https://github.com/apache/tvm/pull/16777. The re-implementation preserves the handling of dynamic shapes specified as python strings, enabling the test cases that were added in https://github.com/apache/tvm/pull/16784. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
