Hello everyone!
I have been trying to export the WavLM model using torch.export but I’m hitting quite a lengthy error that ends with a
torch._dynamo.exc.Unsupported: setattr(UnspecializedNNModuleVariable(Conv1d), weight, ...)
This is apparently due to the torch.nn.utils.weight_norm operator at this line. I tried to workaround it using torch.nn.utils.parametrizations.weight_norm instead, which seems to work. However, this operator is not yet supported by iree-turbine, which is actually what I want to use to compile this model ultimately. So I’m wondering if there are other workarounds to use torch.nn.utils.weight_norm instead, or if this is a known issue that I should raise a GH issue for. I couldn’t find anything mentioning this specific error.
Thanks a lot!