waifu-diffusion-original/data/imagenet_train_hr_indices.p