Pytorch LayerNorm convert to .mlmodel

Open MusickMe opened this issue 7 months ago • 1 comments

❓Question

Pytorch LayerNorm convert to .mlmodel, cann't run in GPU，because it has been broken down into many operators.

May 19 '25 02:05 MusickMe

@MusickMe could you please try to export the model as an ML Program?

model = ct.convert(
source_model, 
convert_to="mlprogram",
compute_precision=ct.precision.FLOAT32)

May 19 '25 19:05 cymbalrush