(*, args, checkpoint_map_location)
| 965 | |
| 966 | |
| 967 | def renderer(*, args, checkpoint_map_location): |
| 968 | print(" loading renderer") |
| 969 | |
| 970 | renderer_checkpoint = torch.load(args.transmitter_checkpoint_path, map_location=checkpoint_map_location) |
| 971 | |
| 972 | renderer_model = renderer_model_from_original_config() |
| 973 | |
| 974 | renderer_diffusers_checkpoint = renderer_model_original_checkpoint_to_diffusers_checkpoint( |
| 975 | renderer_model, renderer_checkpoint |
| 976 | ) |
| 977 | |
| 978 | del renderer_checkpoint |
| 979 | |
| 980 | load_checkpoint_to_model(renderer_diffusers_checkpoint, renderer_model, strict=True) |
| 981 | |
| 982 | print("done loading renderer") |
| 983 | |
| 984 | return renderer_model |
| 985 | |
| 986 | |
| 987 | # prior model will expect clip_mean and clip_std, whic are missing from the state_dict |
no test coverage detected