Fix torch.compile breaking toggle_optimizer / untoggle_optimizer (#21… #1844

Job	Run time
release-version	42s
build-packages / build-packages (lightning)	25s
build-packages / build-packages (fabric)	23s
build-packages / build-packages (pytorch)	28s
build-packages / merge-artifacts	17s
legacy-checkpoints / create-legacy-ckpts	43s
upload-packages	0s
publish-packages	0s
signaling	0s
legacy-checkpoints / add-ckpt-test	0s
	2m 58s

Provide feedback