Function count_conv3d

diffsynth/models/wan_video_vae.py:485–490

(model)

def count_conv3d(model):
    count = 0
    for m in model.modules():
        if check_is_instance(m, CausalConv3d):
            count += 1
    return count


class VideoVAE_(nn.Module):