Closed bogdanteleaga closed 1 year ago
There is probably a better way to do this, but this is temporary to test in the plugin.
Seems like we can't find a much better way to do this with torchscript limitations so I'll merge it as is.
There is probably a better way to do this, but this is temporary to test in the plugin.