SUPIR/__init__.py

SDXL_BASE_CHANNEL_DICT = {
    'cond_output_channels': [320] * 4 + [640] * 3 + [1280] * 3,
    'project_channels': [160] * 4 + [320] * 3 + [640] * 3,
    'concat_channels': [320] * 2 + [640] * 3 + [1280] * 4 + [0]
}

SDXL_REFINE_CHANNEL_DICT = {
    'cond_output_channels': [384] * 4 + [768] * 3 + [1536] * 6,
    'project_channels': [192] * 4 + [384] * 3 + [768] * 6,
    'concat_channels': [384] * 2 + [768] * 3 + [1536] * 7 + [0]
}