Thanks a lot tom. really appreciate it. Yes I know about the jit. but this is a pruned model, so I was kind of searching to see if I could avoid a retraining from scratch or manually creating a custom model definition for it.