Updates paths inside resnets to the new naming scheme (local renaming)
(old_list, n_shave_prefix_segments=0)
| 46 | |
| 47 | # Copied from diffusers.pipelines.stable_diffusion.convert_from_ckpt.renew_vae_resnet_paths |
| 48 | def renew_vae_resnet_paths(old_list, n_shave_prefix_segments=0): |
| 49 | """ |
| 50 | Updates paths inside resnets to the new naming scheme (local renaming) |
| 51 | """ |
| 52 | mapping = [] |
| 53 | for old_item in old_list: |
| 54 | new_item = old_item |
| 55 | |
| 56 | new_item = new_item.replace("nin_shortcut", "conv_shortcut") |
| 57 | new_item = shave_segments(new_item, n_shave_prefix_segments=n_shave_prefix_segments) |
| 58 | |
| 59 | mapping.append({"old": old_item, "new": new_item}) |
| 60 | |
| 61 | return mapping |
| 62 | |
| 63 | |
| 64 | # Copied from diffusers.pipelines.stable_diffusion.convert_from_ckpt.renew_vae_attention_paths |
no test coverage detected