You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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?
The text was updated successfully, but these errors were encountered:
Hi, it's because of the mismatch of the down block and the up block.
The cache_block_id/cache_layer_id we use here is used in the downsampling block/layer. However, the downsampling block and upsampling block that are connected by the skip branch didn't share the same layer_id/block_id. Thus, this part of code functions like a layer/block mapping, and the goal is to find the corresponding layer/block id for this skip branch in the upsampling process
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?
The text was updated successfully, but these errors were encountered: