Fix snapshot restore for abstract shapes and memory kind - #4879

Fix snapshot restore for abstract shapes and memory kind#4879

sadikneipp merged 1 commit into

abhinavsing/snapshot-scaleAI-Hypercomputer/maxtext:abhinavsing/snapshot-scalefrom

ksadi/snapshot-scale-fixAI-Hypercomputer/maxtext:ksadi/snapshot-scale-fixCopy head branch name to clipboard