Spaces:
0x7o
/
Runtime error

hysts HF staff commited on
Commit
0edd51d
1 Parent(s): 9882ef5
Files changed (9) hide show
  1. .gitignore +162 -0
  2. .pre-commit-config.yaml +36 -0
  3. .style.yapf +5 -0
  4. .vscode/settings.json +18 -0
  5. LICENSE +21 -0
  6. README.md +3 -1
  7. app.py +301 -0
  8. requirements.txt +7 -0
  9. style.css +16 -0
.gitignore ADDED
@@ -0,0 +1,162 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ gradio_cached_examples/
2
+
3
+ # Byte-compiled / optimized / DLL files
4
+ __pycache__/
5
+ *.py[cod]
6
+ *$py.class
7
+
8
+ # C extensions
9
+ *.so
10
+
11
+ # Distribution / packaging
12
+ .Python
13
+ build/
14
+ develop-eggs/
15
+ dist/
16
+ downloads/
17
+ eggs/
18
+ .eggs/
19
+ lib/
20
+ lib64/
21
+ parts/
22
+ sdist/
23
+ var/
24
+ wheels/
25
+ share/python-wheels/
26
+ *.egg-info/
27
+ .installed.cfg
28
+ *.egg
29
+ MANIFEST
30
+
31
+ # PyInstaller
32
+ # Usually these files are written by a python script from a template
33
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
34
+ *.manifest
35
+ *.spec
36
+
37
+ # Installer logs
38
+ pip-log.txt
39
+ pip-delete-this-directory.txt
40
+
41
+ # Unit test / coverage reports
42
+ htmlcov/
43
+ .tox/
44
+ .nox/
45
+ .coverage
46
+ .coverage.*
47
+ .cache
48
+ nosetests.xml
49
+ coverage.xml
50
+ *.cover
51
+ *.py,cover
52
+ .hypothesis/
53
+ .pytest_cache/
54
+ cover/
55
+
56
+ # Translations
57
+ *.mo
58
+ *.pot
59
+
60
+ # Django stuff:
61
+ *.log
62
+ local_settings.py
63
+ db.sqlite3
64
+ db.sqlite3-journal
65
+
66
+ # Flask stuff:
67
+ instance/
68
+ .webassets-cache
69
+
70
+ # Scrapy stuff:
71
+ .scrapy
72
+
73
+ # Sphinx documentation
74
+ docs/_build/
75
+
76
+ # PyBuilder
77
+ .pybuilder/
78
+ target/
79
+
80
+ # Jupyter Notebook
81
+ .ipynb_checkpoints
82
+
83
+ # IPython
84
+ profile_default/
85
+ ipython_config.py
86
+
87
+ # pyenv
88
+ # For a library or package, you might want to ignore these files since the code is
89
+ # intended to run in multiple environments; otherwise, check them in:
90
+ # .python-version
91
+
92
+ # pipenv
93
+ # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
94
+ # However, in case of collaboration, if having platform-specific dependencies or dependencies
95
+ # having no cross-platform support, pipenv may install dependencies that don't work, or not
96
+ # install all needed dependencies.
97
+ #Pipfile.lock
98
+
99
+ # poetry
100
+ # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
101
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
102
+ # commonly ignored for libraries.
103
+ # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
104
+ #poetry.lock
105
+
106
+ # pdm
107
+ # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
108
+ #pdm.lock
109
+ # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
110
+ # in version control.
111
+ # https://pdm.fming.dev/#use-with-ide
112
+ .pdm.toml
113
+
114
+ # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
115
+ __pypackages__/
116
+
117
+ # Celery stuff
118
+ celerybeat-schedule
119
+ celerybeat.pid
120
+
121
+ # SageMath parsed files
122
+ *.sage.py
123
+
124
+ # Environments
125
+ .env
126
+ .venv
127
+ env/
128
+ venv/
129
+ ENV/
130
+ env.bak/
131
+ venv.bak/
132
+
133
+ # Spyder project settings
134
+ .spyderproject
135
+ .spyproject
136
+
137
+ # Rope project settings
138
+ .ropeproject
139
+
140
+ # mkdocs documentation
141
+ /site
142
+
143
+ # mypy
144
+ .mypy_cache/
145
+ .dmypy.json
146
+ dmypy.json
147
+
148
+ # Pyre type checker
149
+ .pyre/
150
+
151
+ # pytype static type analyzer
152
+ .pytype/
153
+
154
+ # Cython debug symbols
155
+ cython_debug/
156
+
157
+ # PyCharm
158
+ # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
159
+ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
160
+ # and can be added to the global gitignore or merged into this file. For a more nuclear
161
+ # option (not recommended) you can uncomment the following to ignore the entire idea folder.
162
+ #.idea/
.pre-commit-config.yaml ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ repos:
2
+ - repo: https://github.com/pre-commit/pre-commit-hooks
3
+ rev: v4.2.0
4
+ hooks:
5
+ - id: check-executables-have-shebangs
6
+ - id: check-json
7
+ - id: check-merge-conflict
8
+ - id: check-shebang-scripts-are-executable
9
+ - id: check-toml
10
+ - id: check-yaml
11
+ - id: double-quote-string-fixer
12
+ - id: end-of-file-fixer
13
+ - id: mixed-line-ending
14
+ args: ['--fix=lf']
15
+ - id: requirements-txt-fixer
16
+ - id: trailing-whitespace
17
+ - repo: https://github.com/myint/docformatter
18
+ rev: v1.4
19
+ hooks:
20
+ - id: docformatter
21
+ args: ['--in-place']
22
+ - repo: https://github.com/pycqa/isort
23
+ rev: 5.12.0
24
+ hooks:
25
+ - id: isort
26
+ - repo: https://github.com/pre-commit/mirrors-mypy
27
+ rev: v0.991
28
+ hooks:
29
+ - id: mypy
30
+ args: ['--ignore-missing-imports']
31
+ additional_dependencies: ['types-python-slugify']
32
+ - repo: https://github.com/google/yapf
33
+ rev: v0.32.0
34
+ hooks:
35
+ - id: yapf
36
+ args: ['--parallel', '--in-place']
.style.yapf ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ [style]
2
+ based_on_style = pep8
3
+ blank_line_before_nested_class_or_def = false
4
+ spaces_before_comment = 2
5
+ split_before_logical_operator = true
.vscode/settings.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "python.linting.enabled": true,
3
+ "python.linting.flake8Enabled": true,
4
+ "python.linting.pylintEnabled": false,
5
+ "python.linting.lintOnSave": true,
6
+ "python.formatting.provider": "yapf",
7
+ "python.formatting.yapfArgs": [
8
+ "--style={based_on_style: pep8, indent_width: 4, blank_line_before_nested_class_or_def: false, spaces_before_comment: 2, split_before_logical_operator: true}"
9
+ ],
10
+ "[python]": {
11
+ "editor.formatOnType": true,
12
+ "editor.codeActionsOnSave": {
13
+ "source.organizeImports": true
14
+ }
15
+ },
16
+ "editor.formatOnSave": true,
17
+ "files.insertFinalNewline": true
18
+ }
LICENSE ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MIT License
2
+
3
+ Copyright (c) 2023 hysts
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
README.md CHANGED
@@ -1,12 +1,14 @@
1
  ---
2
- title: SD XL
3
  emoji: 🌍
4
  colorFrom: gray
5
  colorTo: purple
6
  sdk: gradio
7
  sdk_version: 3.39.0
8
  app_file: app.py
 
9
  pinned: false
 
10
  ---
11
 
12
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
+ title: SD-XL
3
  emoji: 🌍
4
  colorFrom: gray
5
  colorTo: purple
6
  sdk: gradio
7
  sdk_version: 3.39.0
8
  app_file: app.py
9
+ license: mit
10
  pinned: false
11
+ suggested_hardware: a10g-small
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
app.py ADDED
@@ -0,0 +1,301 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python
2
+
3
+ from __future__ import annotations
4
+
5
+ import os
6
+ import random
7
+
8
+ import gradio as gr
9
+ import numpy as np
10
+ import PIL.Image
11
+ import torch
12
+ from diffusers import DiffusionPipeline
13
+
14
+ DESCRIPTION = '# SD-XL'
15
+ if not torch.cuda.is_available():
16
+ DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'
17
+
18
+ MAX_SEED = np.iinfo(np.int32).max
19
+ CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv(
20
+ 'CACHE_EXAMPLES') == '1'
21
+ MAX_IMAGE_SIZE = int(os.getenv('MAX_IMAGE_SIZE', '1024'))
22
+ USE_TORCH_COMPILE = os.getenv('USE_TORCH_COMPILE') == '1'
23
+ ENABLE_CPU_OFFLOAD = os.getenv('ENABLE_CPU_OFFLOAD') == '1'
24
+
25
+ device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
26
+ if torch.cuda.is_available():
27
+ pipe = DiffusionPipeline.from_pretrained(
28
+ 'stabilityai/stable-diffusion-xl-base-1.0',
29
+ torch_dtype=torch.float16,
30
+ use_safetensors=True,
31
+ variant='fp16')
32
+ refiner = DiffusionPipeline.from_pretrained(
33
+ 'stabilityai/stable-diffusion-xl-refiner-1.0',
34
+ torch_dtype=torch.float16,
35
+ use_safetensors=True,
36
+ variant='fp16')
37
+
38
+ if ENABLE_CPU_OFFLOAD:
39
+ pipe.enable_model_cpu_offload()
40
+ refiner.enable_model_cpu_offload()
41
+ else:
42
+ pipe.to(device)
43
+ refiner.to(device)
44
+
45
+ if USE_TORCH_COMPILE:
46
+ pipe.unet = torch.compile(pipe.unet,
47
+ mode='reduce-overhead',
48
+ fullgraph=True)
49
+ else:
50
+ pipe = None
51
+ refiner = None
52
+
53
+
54
+ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
55
+ if randomize_seed:
56
+ seed = random.randint(0, MAX_SEED)
57
+ return seed
58
+
59
+
60
+ def generate(prompt: str,
61
+ negative_prompt: str = '',
62
+ prompt_2: str = '',
63
+ negative_prompt_2: str = '',
64
+ use_negative_prompt: bool = False,
65
+ use_prompt_2: bool = False,
66
+ use_negative_prompt_2: bool = False,
67
+ seed: int = 0,
68
+ width: int = 1024,
69
+ height: int = 1024,
70
+ guidance_scale_base: float = 5.0,
71
+ guidance_scale_refiner: float = 5.0,
72
+ num_inference_steps_base: int = 50,
73
+ num_inference_steps_refiner: int = 50,
74
+ apply_refiner: bool = False) -> PIL.Image.Image:
75
+ generator = torch.Generator().manual_seed(seed)
76
+
77
+ if not use_negative_prompt:
78
+ negative_prompt = None # type: ignore
79
+ if not use_prompt_2:
80
+ prompt_2 = None # type: ignore
81
+ if not use_negative_prompt_2:
82
+ negative_prompt_2 = None # type: ignore
83
+
84
+ if not apply_refiner:
85
+ return pipe(prompt=prompt,
86
+ negative_prompt=negative_prompt,
87
+ prompt_2=prompt_2,
88
+ negative_prompt_2=negative_prompt_2,
89
+ width=width,
90
+ height=height,
91
+ guidance_scale=guidance_scale_base,
92
+ num_inference_steps=num_inference_steps_base,
93
+ generator=generator,
94
+ output_type='pil').images[0]
95
+ else:
96
+ latents = pipe(prompt=prompt,
97
+ negative_prompt=negative_prompt,
98
+ prompt_2=prompt_2,
99
+ negative_prompt_2=negative_prompt_2,
100
+ width=width,
101
+ height=height,
102
+ guidance_scale=guidance_scale_base,
103
+ num_inference_steps=num_inference_steps_base,
104
+ generator=generator,
105
+ output_type='latent').images
106
+ image = refiner(prompt=prompt,
107
+ negative_prompt=negative_prompt,
108
+ prompt_2=prompt_2,
109
+ negative_prompt_2=negative_prompt_2,
110
+ guidance_scale=guidance_scale_refiner,
111
+ num_inference_steps=num_inference_steps_refiner,
112
+ image=latents,
113
+ generator=generator).images[0]
114
+ return image
115
+
116
+
117
+ examples = [
118
+ 'Astronaut in a jungle, cold color palette, muted colors, detailed, 8k',
119
+ 'An astronaut riding a green horse',
120
+ ]
121
+
122
+ with gr.Blocks(css='style.css') as demo:
123
+ gr.Markdown(DESCRIPTION)
124
+ gr.DuplicateButton(value='Duplicate Space for private use',
125
+ elem_id='duplicate-button',
126
+ visible=os.getenv('SHOW_DUPLICATE_BUTTON') == '1')
127
+ with gr.Box():
128
+ with gr.Row():
129
+ prompt = gr.Text(
130
+ label='Prompt',
131
+ show_label=False,
132
+ max_lines=1,
133
+ placeholder='Enter your prompt',
134
+ container=False,
135
+ )
136
+ run_button = gr.Button('Run', scale=0)
137
+ result = gr.Image(label='Result', show_label=False)
138
+ with gr.Accordion('Advanced options', open=False):
139
+ with gr.Row():
140
+ use_negative_prompt = gr.Checkbox(label='Use negative prompt',
141
+ value=False)
142
+ use_prompt_2 = gr.Checkbox(label='Use prompt 2', value=False)
143
+ use_negative_prompt_2 = gr.Checkbox(
144
+ label='Use negative prompt 2', value=False)
145
+ negative_prompt = gr.Text(
146
+ label='Negative prompt',
147
+ max_lines=1,
148
+ placeholder='Enter a negative prompt',
149
+ visible=False,
150
+ )
151
+ prompt_2 = gr.Text(
152
+ label='Prompt 2',
153
+ max_lines=1,
154
+ placeholder='Enter your prompt',
155
+ visible=False,
156
+ )
157
+ negative_prompt_2 = gr.Text(
158
+ label='Negative prompt 2',
159
+ max_lines=1,
160
+ placeholder='Enter a negative prompt',
161
+ visible=False,
162
+ )
163
+
164
+ seed = gr.Slider(label='Seed',
165
+ minimum=0,
166
+ maximum=MAX_SEED,
167
+ step=1,
168
+ value=0)
169
+ randomize_seed = gr.Checkbox(label='Randomize seed', value=True)
170
+ with gr.Row():
171
+ width = gr.Slider(
172
+ label='Width',
173
+ minimum=256,
174
+ maximum=MAX_IMAGE_SIZE,
175
+ step=32,
176
+ value=1024,
177
+ )
178
+ height = gr.Slider(
179
+ label='Height',
180
+ minimum=256,
181
+ maximum=MAX_IMAGE_SIZE,
182
+ step=32,
183
+ value=1024,
184
+ )
185
+ apply_refiner = gr.Checkbox(label='Apply refiner', value=False)
186
+ with gr.Row():
187
+ guidance_scale_base = gr.Slider(
188
+ label='Guidance scale for base',
189
+ minimum=1,
190
+ maximum=20,
191
+ step=0.1,
192
+ value=5.0)
193
+ num_inference_steps_base = gr.Slider(
194
+ label='Number of inference steps for base',
195
+ minimum=10,
196
+ maximum=100,
197
+ step=1,
198
+ value=50)
199
+ with gr.Row(visible=False) as refiner_params:
200
+ guidance_scale_refiner = gr.Slider(
201
+ label='Guidance scale for refiner',
202
+ minimum=1,
203
+ maximum=20,
204
+ step=0.1,
205
+ value=5.0)
206
+ num_inference_steps_refiner = gr.Slider(
207
+ label='Number of inference steps for refiner',
208
+ minimum=10,
209
+ maximum=100,
210
+ step=1,
211
+ value=50)
212
+
213
+ gr.Examples(examples=examples,
214
+ inputs=prompt,
215
+ outputs=result,
216
+ fn=generate,
217
+ cache_examples=CACHE_EXAMPLES)
218
+
219
+ use_negative_prompt.change(
220
+ fn=lambda x: gr.update(visible=x),
221
+ inputs=use_negative_prompt,
222
+ outputs=negative_prompt,
223
+ queue=False,
224
+ api_name=False,
225
+ )
226
+ use_prompt_2.change(
227
+ fn=lambda x: gr.update(visible=x),
228
+ inputs=use_prompt_2,
229
+ outputs=prompt_2,
230
+ queue=False,
231
+ api_name=False,
232
+ )
233
+ use_negative_prompt_2.change(
234
+ fn=lambda x: gr.update(visible=x),
235
+ inputs=use_negative_prompt_2,
236
+ outputs=negative_prompt_2,
237
+ queue=False,
238
+ api_name=False,
239
+ )
240
+ apply_refiner.change(
241
+ fn=lambda x: gr.update(visible=x),
242
+ inputs=apply_refiner,
243
+ outputs=refiner_params,
244
+ queue=False,
245
+ api_name=False,
246
+ )
247
+
248
+ inputs = [
249
+ prompt,
250
+ negative_prompt,
251
+ prompt_2,
252
+ negative_prompt_2,
253
+ use_negative_prompt,
254
+ use_prompt_2,
255
+ use_negative_prompt_2,
256
+ seed,
257
+ width,
258
+ height,
259
+ guidance_scale_base,
260
+ guidance_scale_refiner,
261
+ num_inference_steps_base,
262
+ num_inference_steps_refiner,
263
+ apply_refiner,
264
+ ]
265
+ prompt.submit(
266
+ fn=randomize_seed_fn,
267
+ inputs=[seed, randomize_seed],
268
+ outputs=seed,
269
+ queue=False,
270
+ api_name=False,
271
+ ).then(
272
+ fn=generate,
273
+ inputs=inputs,
274
+ outputs=result,
275
+ api_name='run',
276
+ )
277
+ negative_prompt.submit(
278
+ fn=randomize_seed_fn,
279
+ inputs=[seed, randomize_seed],
280
+ outputs=seed,
281
+ queue=False,
282
+ api_name=False,
283
+ ).then(
284
+ fn=generate,
285
+ inputs=inputs,
286
+ outputs=result,
287
+ api_name=False,
288
+ )
289
+ run_button.click(
290
+ fn=randomize_seed_fn,
291
+ inputs=[seed, randomize_seed],
292
+ outputs=seed,
293
+ queue=False,
294
+ api_name=False,
295
+ ).then(
296
+ fn=generate,
297
+ inputs=inputs,
298
+ outputs=result,
299
+ api_name=False,
300
+ )
301
+ demo.queue(max_size=20).launch()
requirements.txt ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ accelerate==0.21.0
2
+ diffusers==0.19.0
3
+ gradio==3.39.0
4
+ invisible-watermark==0.2.0
5
+ Pillow==10.0.0
6
+ torch==2.0.1
7
+ transformers==4.31.0
style.css ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ h1 {
2
+ text-align: center;
3
+ }
4
+
5
+ #duplicate-button {
6
+ margin: auto;
7
+ color: #fff;
8
+ background: #1565c0;
9
+ border-radius: 100vh;
10
+ }
11
+
12
+ #component-0 {
13
+ max-width: 730px;
14
+ margin: auto;
15
+ padding-top: 1.5rem;
16
+ }