Fix regression in the recent model saving refactor. Pass the non unet pieces down the layers so that checkpoints are complete.
71 KiB
71 KiB
Fix regression in the recent model saving refactor. Pass the non unet pieces down the layers so that checkpoints are complete.