(state_dict)
| 261 | |
| 262 | |
| 263 | def get_caption_projection_dim(state_dict): |
| 264 | caption_projection_dim = state_dict["context_embedder.weight"].shape[0] |
| 265 | return caption_projection_dim |
| 266 | |
| 267 | |
| 268 | def main(args): |
no outgoing calls
no test coverage detected
searching dependent graphs…