-
Notifications
You must be signed in to change notification settings - Fork 0
/
modelsListPA.py
26 lines (24 loc) · 1.09 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
25
26
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-900m-1024-ft",
"TensorFamily/SigmaJourney",
"ptx0/pixart-900m-1024-ft-v0.7-stage1",
# "frutiemax/TwistedReality-pixart-1024ms",
# "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