File size: 8,721 Bytes
b659c17
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a17b715
b659c17
 
 
 
a17b715
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b659c17
 
 
 
 
 
 
 
a17b715
 
 
 
b659c17
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d29036f
acb522c
b659c17
a17b715
b659c17
a17b715
b659c17
 
 
 
 
 
 
 
 
 
 
 
a17b715
b659c17
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a17b715
b659c17
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
---
license: creativeml-openrail-m
base_model: "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS"
tags:
  - stable-diffusion
  - stable-diffusion-diffusers
  - text-to-image
  - diffusers
  - full

inference: true
widget:
- text: 'unconditional (blank prompt)'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_0_0.png
- text: 'a woman sitting on the grass'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_1_0.png
- text: 'a professional photo headshot of a man in studio lighting'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_2_0.png
- text: 'a person holding a sign that reads ''SOON'''
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_3_0.png
- text: 'Alien marketplace, bizarre creatures, exotic goods, vibrant colors, otherworldly atmosphere'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_4_0.png
- text: 'Child holding a balloon, happy expression, colorful balloons, sunny day, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_5_0.png
- text: 'a 4-panel comic strip showing an orange cat saying the words ''HELP'' and ''LASAGNA'''
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_6_0.png
- text: 'a hand is holding a comic book with a cover that reads ''The Adventures of Superhero'''
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_7_0.png
- text: 'Underground cave filled with crystals, glowing lights, reflective surfaces, fantasy environment, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_8_0.png
- text: 'Bustling cyberpunk bazaar, vendors, neon signs, advanced tech, crowded, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_9_0.png
- text: 'Cyberpunk hacker in a dark room, neon glow, multiple screens, intense focus, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_10_0.png
- text: 'a cybernetic anne of green gables with neural implant and bio mech augmentations'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_11_0.png
- text: 'Post-apocalyptic cityscape, ruined buildings, overgrown vegetation, dark and gritty, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_12_0.png
- text: 'Magical castle in a lush forest, glowing windows, fantasy architecture, high resolution, detailed textures'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_13_0.png
- text: 'Ruins of an ancient temple in an enchanted forest, glowing runes, mystical creatures, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_14_0.png
- text: 'Mystical forest, glowing plants, fairies, magical creatures, fantasy art, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_15_0.png
- text: 'Magical garden with glowing flowers, fairies, serene atmosphere, detailed plants, high resolution'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_16_0.png
- text: 'Whimsical garden filled with fairies, magical plants, sparkling lights, serene atmosphere, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_17_0.png
- text: 'Majestic dragon soaring through the sky, detailed scales, dynamic pose, fantasy art, high resolution'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_18_0.png
- text: 'Fantasy world, floating islands in the sky, waterfalls, lush vegetation, detailed landscape, high resolution'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_19_0.png
- text: 'Futuristic city skyline at night, neon lights, cyberpunk style, high contrast, sharp focus'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_20_0.png
- text: 'Space battle scene, starships fighting, laser beams, explosions, cosmic background'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_21_0.png
- text: 'Abandoned fairground at night, eerie rides, ghostly figures, fog, dark atmosphere, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_22_0.png
- text: 'Spooky haunted mansion on a hill, dark and eerie, glowing windows, ghostly atmosphere, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_23_0.png
- text: 'a hardcover physics textbook that is called PHYSICS FOR DUMMIES'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_24_0.png
- text: 'Epic medieval battle, knights in armor, dynamic action, detailed landscape, high resolution'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_25_0.png
- text: 'Bustling medieval market with merchants, knights, and jesters, vibrant colors, detailed'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_26_0.png
- text: 'Cozy medieval tavern, warm firelight, adventurers drinking, detailed interior, rustic atmosphere'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_27_0.png
- text: 'Futuristic city skyline at night, neon lights, cyberpunk style, high contrast, sharp focus'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_28_0.png
- text: 'Forest with neon-lit trees, glowing plants, bioluminescence, surreal atmosphere, high detail'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_29_0.png
- text: 'Bright neon sign in a busy city street, ''Open 24 Hours'', bold typography, glowing lights'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_30_0.png
- text: 'Retro diner sign, ''Joe''s Diner'', classic 1950s design, neon lights, weathered look'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_31_0.png
- text: 'Vintage store sign with elaborate typography, ''Antique Shop'', hand-painted, weathered look'
  parameters:
    negative_prompt: 'blurry, cropped, ugly'
  output:
    url: ./assets/image_32_0.png
---

# pixart-training

This is a full rank finetune derived from [PixArt-alpha/PixArt-Sigma-XL-2-1024-MS](https://huggingface.co/PixArt-alpha/PixArt-Sigma-XL-2-1024-MS).



No validation prompt was used during training.


None


## Validation settings
- CFG: `7.5`
- CFG Rescale: `0.0`
- Steps: `30`
- Sampler: `euler`
- Seed: `42`
- Resolution: `1024`

Note: The validation settings are not necessarily the same as the [training settings](#training-settings).

You can find some example images in the following gallery:


<Gallery />

The text encoder **was not** trained.
You may reuse the base model text encoder for inference.


## Training settings

- Training epochs: 5
- Training steps: 6000
- Learning rate: 8e-06
- Effective batch size: 128
  - Micro-batch size: 32
  - Gradient accumulation steps: 4
  - Number of GPUs: 1
- Prediction type: epsilon
- Rescaled betas zero SNR: False
- Optimizer: AdamW, stochastic bf16
- Precision: Pure BF16
- Xformers: Enabled


## Datasets

### mj-v6
- Repeats: 0
- Total number of images: 134144
- Total number of aspect buckets: 1
- Resolution: 1.0 megapixels
- Cropped: False
- Crop style: None
- Crop aspect: None


## Inference


```python
import torch
from diffusers import DiffusionPipeline



model_id = "pixart-training"
prompt = "An astronaut is riding a horse through the jungles of Thailand."
negative_prompt = "malformed, disgusting, overexposed, washed-out"

pipeline = DiffusionPipeline.from_pretrained(model_id)
pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
image = pipeline(
    prompt=prompt,
    negative_prompt='blurry, cropped, ugly',
    num_inference_steps=30,
    generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
    width=1152,
    height=768,
    guidance_scale=7.5,
    guidance_rescale=0.0,
).images[0]
image.save("output.png", format="PNG")
```