Open to ideas on if this is an actual missing issue or something I can solve.
RuntimeError                              Traceback (most recent call last)
 in ()
1 kp_driving = kp_detector(image_tensor)
2 traced_generator = torch.jit.trace(generator, (image_tensor, kp_driving, kp_driving), strict=False)
----> 3 optimized_traced_generator = optimize_for_mobile(traced_generator, backend=‘metal’)
4 optimized_traced_generator._save_for_lite_interpreter("/content/generator_metal.pt")
/usr/local/lib/python3.7/dist-packages/torch/utils/mobile_optimizer.py in optimize_for_mobile(script_module, optimization_blocklist, preserved_methods, backend)
67         optimized_cpp_module = torch._C._jit_pass_vulkan_optimize_for_mobile(script_module._c, preserved_methods_str)
68     elif backend == ‘metal’:
—> 69         optimized_cpp_module = torch._C._jit_pass_metal_optimize_for_mobile(script_module._c, preserved_methods_str)
70     else:
71         raise TypeError(“Unknown backend, must be one of ‘CPU’, ‘Vulkan’ or ‘Metal’”)
RuntimeError: 0INTERNAL ASSERT FAILED at “…/torch/csrc/jit/ir/alias_analysis.cpp”:584, please report a bug to PyTorch. We don’t have an op for metal_prepack::conv2d_prepack but it isn’t a special case.  Argument types: Tensor, Tensor, int[], int[], int[], int, NoneType, NoneType,