Inits the start of jit support. Currently the language models can't be pushed to jit to my knowledge, and neither can unet currently. However in the newest version we plan to fully use timm, which is, so I'm not concerned about the unet. This should be changed once the timm support is finalized to show a unet example.
Inits the start of jit support. Currently the language models can't be pushed to jit to my knowledge, and neither can unet currently. However in the newest version we plan to fully use timm, which is, so I'm not concerned about the unet. This should be changed once the timm support is finalized to show a unet example.