Fabrice-TIERCELIN commited on
Commit
85a4eb3
1 Parent(s): e8243c3

Upload __init__.py

Browse files
Files changed (1) hide show
  1. SUPIR/__init__.py +11 -0
SUPIR/__init__.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ SDXL_BASE_CHANNEL_DICT = {
2
+ 'cond_output_channels': [320] * 4 + [640] * 3 + [1280] * 3,
3
+ 'project_channels': [160] * 4 + [320] * 3 + [640] * 3,
4
+ 'concat_channels': [320] * 2 + [640] * 3 + [1280] * 4 + [0]
5
+ }
6
+
7
+ SDXL_REFINE_CHANNEL_DICT = {
8
+ 'cond_output_channels': [384] * 4 + [768] * 3 + [1536] * 6,
9
+ 'project_channels': [192] * 4 + [384] * 3 + [768] * 6,
10
+ 'concat_channels': [384] * 2 + [768] * 3 + [1536] * 7 + [0]
11
+ }