Add new 900M config for the traditional loader, might be better to have depth selectable?

2024-06-23 19:48:34 +01:00
parent 8f42946305
commit 46e4fc8ce3
1 changed files with 15 additions and 0 deletions
@@ -37,6 +37,21 @@ pixart_conf = {
 		},
 		"sampling_settings" : sampling_settings,
 	},
 	"PixArtMS_Sigma_XL_2_900M": {
 		"target": "PixArtMSSigma",
 		"unet_config": {
 			"input_size": 1024 // 8,
 			"token_num": 300,
 			"depth": 42,
 			"num_heads": 16,
 			"patch_size": 2,
 			"hidden_size": 1152,
 			"micro_condition": False,
 			"pe_interpolation": 2,
 			"model_max_length": 300,
 		},
 		"sampling_settings": sampling_settings,
 	},
 	"PixArtMS_Sigma_XL_2_2K": {
 		"target": "PixArtMSSigma",
 		"unet_config": {