Caching Mechanism for choosing skip branch

[Kyuseok-nam]

Hi, I was going over the code for DeepCache and was quite confused with a particular part of the code where you choose the skip branch in the unet model.

from unet_2d_condition.py file, line 1102~1108, Why did you implment this if condition? it seems to limit access of other skip branch or layers.

Also if I'm understanding correctly, Cache_layer_id : indicates which big block to choose from ex) CrossAttn2Dblock, DownBlock2D,... Cache_block_id: indicates which block (small blocks inside the big block) to choose from ex) transformer2dmodel, resnet

is this correct?