-
Notifications
You must be signed in to change notification settings - Fork 0
/
modelsListPA.py
24 lines (22 loc) · 1.08 KB
/
modelsListPA.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
models_list_alpha = [
"PixArt-alpha/PixArt-XL-2-256x256", # smol
"PixArt-alpha/PixArt-XL-2-512x512",
"PixArt-alpha/PixArt-XL-2-1024-MS",
"jasperai/flash-pixart",
"PixArt-alpha/PixArt-LCM-XL-2-1024-MS", # flash-pixart better
"PixArt-alpha/PixArt-Alpha-DMD-XL-2-512x512", # flash-pixart better
# "artificialguybr/Fascinatio-PixartAlpha1024-Finetuned",
]
models_list_sigma = [
"PixArt-alpha/PixArt-Sigma-XL-2-256x256", # smol
"PixArt-alpha/PixArt-Sigma-XL-2-512-MS",
"PixArt-alpha/PixArt-Sigma-XL-2-1024-MS",
"PixArt-alpha/PixArt-Sigma-XL-2-2K-MS", # undertrained
"ptx0/pixart-sigma",
# "ptx0/pixart-reality-mix", # not working with standard pipelines, here for future, v-pred?
# "frutiemax/VintageKnockers-Pixart-Sigma-XL-2-512-MS", # nsfw
# "frutiemax/VintageKnockers-Pixart-Sigma-XL-2-1024-MS", # nsfw
]
defaultModel = models_list_sigma[2]
defaultWidth = 1024
defaultHeight = 1024