(index, token, cross_maps, res, attn16, out)
| 404 | out_new3, out_subject3 = {0:[], 1:[]}, {0:[], 1:[]} |
| 405 | |
| 406 | def process_token(index, token, cross_maps, res, attn16, out): |
| 407 | if attn16: |
| 408 | cross_maps_ = cross_maps[index][:, :, :, token] |
| 409 | else: |
| 410 | cross_maps_ = cross_maps[index][:, :, :, token].reshape(1, cross_maps.shape[1], res, res) |
| 411 | cross_maps_ = F.interpolate(cross_maps_, size=(64, 64), mode='bicubic').reshape(cross_maps.shape[1], 64, 64) |
| 412 | out[index].append(cross_maps_) |
| 413 | |
| 414 | for _, value in self.controller.step_store.items(): |
| 415 | for item in value: |
nothing calls this directly
no outgoing calls
no test coverage detected