(args)
| 7 | |
| 8 | |
| 9 | def load_original_state_dict(args): |
| 10 | model_pt = hf_hub_download(repo_id=args.original_state_dict_repo_id, filename="aura_diffusion_pytorch_model.bin") |
| 11 | state_dict = torch.load(model_pt, map_location="cpu") |
| 12 | return state_dict |
| 13 | |
| 14 | |
| 15 | def calculate_layers(state_dict_keys, key_prefix): |
no test coverage detected
searching dependent graphs…