diff --git a/PixArt/conf.py b/PixArt/conf.py index 7602dcd..86c1d7d 100644 --- a/PixArt/conf.py +++ b/PixArt/conf.py @@ -37,6 +37,21 @@ pixart_conf = { }, "sampling_settings" : sampling_settings, }, + "PixArtMS_Sigma_XL_2_2K": { + "target": "PixArtMSSigma", + "unet_config": { + "input_size" : 2048//8, + "token_num" : 300, + "depth" : 28, + "num_heads" : 16, + "patch_size" : 2, + "hidden_size" : 1152, + "micro_condition": False, + "pe_interpolation": 4, + "model_max_length": 300, + }, + "sampling_settings" : sampling_settings, + }, "PixArt_XL_2": { # models/PixArt "target": "PixArt", "unet_config": { @@ -90,6 +105,18 @@ pixart_res = { '2.50': [800, 320], '2.89': [832, 288], '3.00': [864, 288], '3.11': [896, 288], '3.62': [928, 256], '3.75': [960, 256], '3.88': [992, 256], '4.00': [1024,256] }, + "PixArtMS_Sigma_XL_2_2K": { + '0.25': [1024, 4096], '0.26': [1024, 3968], '0.27': [1024, 3840], '0.28': [1024, 3712], + '0.32': [1152, 3584], '0.33': [1152, 3456], '0.35': [1152, 3328], '0.40': [1280, 3200], + '0.42': [1280, 3072], '0.48': [1408, 2944], '0.50': [1408, 2816], '0.52': [1408, 2688], + '0.57': [1536, 2688], '0.60': [1536, 2560], '0.68': [1664, 2432], '0.72': [1664, 2304], + '0.78': [1792, 2304], '0.82': [1792, 2176], '0.88': [1920, 2176], '0.94': [1920, 2048], + '1.00': [2048, 2048], '1.07': [2048, 1920], '1.13': [2176, 1920], '1.21': [2176, 1792], + '1.29': [2304, 1792], '1.38': [2304, 1664], '1.46': [2432, 1664], '1.67': [2560, 1536], + '1.75': [2688, 1536], '2.00': [2816, 1408], '2.09': [2944, 1408], '2.40': [3072, 1280], + '2.50': [3200, 1280], '2.89': [3328, 1152], '3.00': [3456, 1152], '3.11': [3584, 1152], + '3.62': [3712, 1024], '3.75': [3840, 1024], '3.88': [3968, 1024], '4.00': [4096, 1024] + } } # These should be the same pixart_res.update({