Yifan Shen
Yifan Shen
Hey folks, a potential source of `pythonop` is PyTorch autograd. Concretely, if you customize something with `torch.autograd.Function`, then PyTorch autograd may give you some `pythonop` that cannot be converted. If...
Hey @polvalente, could you please elaborate your use case a little bit? 1. How do you get your StableHLO module in the first place? (By training a JAX model I...
I see, since `XlaCallModule` is a delegation, unfortunately we won't be able to convert it as is. However, I do think if you can let TF graph digest the StableHLO...
Hey @FreakTheMighty, I cannot reproduce your issue. Could you please share: 1. A full (but minimal) tunable script? 2. Your tensorflow and coremltools version?
Also, please try our [Create ML](https://developer.apple.com/machine-learning/create-ml/) as a workaround while we look into this issue, thanks!
Hey @antmikinka, would this simpler export work for you? ``` python -m examples.models.llama2.export_llama --checkpoint /Users/anthonymikinka/executorch/llama-2-7b-chat/consolidated.00.pth --params /Users/anthonymikinka/executorch/llama-2-7b-chat/params.json -kv --coreml ``` Concretely, this is a good start point that we have...
Hi @y-ich, in order for us to look into the issue, could you please create a minimal reproduce? Concretely, a very small torch model like ``` class Model(torch.nn.Module): def forward(self,...
Thanks for the reproduce! Yes, it reproduces on my end as well. I think instead of that assert https://github.com/apple/coremltools/blob/0b713d23ba86fa6cdb23e8b24e9e65e0f8c35807/coremltools/converters/mil/frontend/torch/ops.py#L1615 We should append a cast to the conversion https://github.com/apple/coremltools/blob/0b713d23ba86fa6cdb23e8b24e9e65e0f8c35807/coremltools/converters/mil/frontend/torch/ops.py#L1616-L1621 ``` if...
@junpeiz We managed to support some complex algebra by decomposing the real and the imaginary parts. Is it possible to do the same for `slice_by_index`? I guess not, since 1....
Hi @VishalBCA-ops, many thanks for bringing the awesome kosmos model to our attention! I think this model is in PyTorch format, so probably you can start with trying our [PyTorch...