justmalhar Linaqruf commited on
Commit
a4e562b
0 Parent(s):

Duplicate from Linaqruf/Animagine-XL

Browse files

Co-authored-by: Linaqruf <Linaqruf@users.noreply.huggingface.co>

Files changed (10) hide show
  1. .gitattributes +35 -0
  2. .gitignore +162 -0
  3. .pre-commit-config.yaml +36 -0
  4. .style.yapf +5 -0
  5. .vscode/settings.json +18 -0
  6. LICENSE +21 -0
  7. README.md +17 -0
  8. app.py +329 -0
  9. requirements.txt +7 -0
  10. style.css +25 -0
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
.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 ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: Animagine-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
+ duplicated_from: Linaqruf/Animagine-XL
13
+ ---
14
+
15
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
16
+
17
+ https://arxiv.org/abs/2307.01952
app.py ADDED
@@ -0,0 +1,329 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.models import AutoencoderKL
13
+ from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
14
+
15
+ DESCRIPTION = '# Animagine XL'
16
+ if not torch.cuda.is_available():
17
+ DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'
18
+
19
+ MAX_SEED = np.iinfo(np.int32).max
20
+ CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv(
21
+ 'CACHE_EXAMPLES') == '1'
22
+ MAX_IMAGE_SIZE = int(os.getenv('MAX_IMAGE_SIZE', '2048'))
23
+ USE_TORCH_COMPILE = os.getenv('USE_TORCH_COMPILE') == '1'
24
+ ENABLE_CPU_OFFLOAD = os.getenv('ENABLE_CPU_OFFLOAD') == '1'
25
+
26
+ MODEL = "Linaqruf/animagine-xl"
27
+
28
+ device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
29
+ if torch.cuda.is_available():
30
+ pipe = StableDiffusionXLPipeline.from_pretrained(
31
+ MODEL,
32
+ torch_dtype=torch.float16,
33
+ use_safetensors=True,
34
+ variant='fp16')
35
+
36
+ pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
37
+
38
+ if ENABLE_CPU_OFFLOAD:
39
+ pipe.enable_model_cpu_offload()
40
+ else:
41
+ pipe.to(device)
42
+
43
+ if USE_TORCH_COMPILE:
44
+ pipe.unet = torch.compile(pipe.unet,
45
+ mode='reduce-overhead',
46
+ fullgraph=True)
47
+ else:
48
+ pipe = None
49
+
50
+
51
+ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
52
+ if randomize_seed:
53
+ seed = random.randint(0, MAX_SEED)
54
+ return seed
55
+
56
+
57
+ def generate(prompt: str,
58
+ negative_prompt: str = '',
59
+ prompt_2: str = '',
60
+ negative_prompt_2: str = '',
61
+ use_prompt_2: bool = False,
62
+ seed: int = 0,
63
+ width: int = 1024,
64
+ height: int = 1024,
65
+ target_width: int = 1024,
66
+ target_height: int = 1024,
67
+ original_width: int = 4096,
68
+ original_height: int = 4096,
69
+ guidance_scale_base: float = 12.0,
70
+ num_inference_steps_base: int = 50) -> PIL.Image.Image:
71
+
72
+ generator = torch.Generator().manual_seed(seed)
73
+
74
+ if negative_prompt == '':
75
+ negative_prompt = None # type: ignore
76
+ if not use_prompt_2:
77
+ prompt_2 = None # type: ignore
78
+ negative_prompt_2 = None # type: ignore
79
+ if negative_prompt_2 == '':
80
+ negative_prompt_2 = None
81
+
82
+ return pipe(prompt=prompt,
83
+ negative_prompt=negative_prompt,
84
+ prompt_2=prompt_2,
85
+ negative_prompt_2=negative_prompt_2,
86
+ width=width,
87
+ height=height,
88
+ target_size=(target_width, target_height),
89
+ original_size=(original_width, original_height),
90
+ guidance_scale=guidance_scale_base,
91
+ num_inference_steps=num_inference_steps_base,
92
+ generator=generator,
93
+ output_type='pil').images[0]
94
+
95
+
96
+ examples = [
97
+ 'face focus, cute, masterpiece, best quality, 1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, night, turtleneck',
98
+ 'face focus, bishounen, masterpiece, best quality, 1boy, green hair, sweater, looking at viewer, upper body, beanie, outdoors, night, turtleneck',
99
+ ]
100
+
101
+ # choices = [
102
+ # "Vertical (9:16)",
103
+ # "Portrait (4:5)",
104
+ # "Square (1:1)",
105
+ # "Photo (4:3)",
106
+ # "Landscape (3:2)",
107
+ # "Widescreen (16:9)",
108
+ # "Cinematic (21:9)",
109
+ # ]
110
+
111
+ # choice_to_size = {
112
+ # "Vertical (9:16)": (768, 1344),
113
+ # "Portrait (4:5)": (912, 1144),
114
+ # "Square (1:1)": (1024, 1024),
115
+ # "Photo (4:3)": (1184, 888),
116
+ # "Landscape (3:2)": (1256, 832),
117
+ # "Widescreen (16:9)": (1368, 768),
118
+ # "Cinematic (21:9)": (1568, 672),
119
+ # }
120
+
121
+ with gr.Blocks(css='style.css') as demo:
122
+ gr.Markdown(DESCRIPTION)
123
+ gr.DuplicateButton(value='Duplicate Space for private use',
124
+ elem_id='duplicate-button',
125
+ visible=os.getenv('SHOW_DUPLICATE_BUTTON') == '1')
126
+ with gr.Row():
127
+ with gr.Column(scale=1):
128
+ prompt = gr.Text(
129
+ label='Prompt',
130
+ max_lines=1,
131
+ placeholder='Enter your prompt',
132
+ )
133
+ negative_prompt = gr.Text(
134
+ label='Negative Prompt',
135
+ max_lines=1,
136
+ placeholder='Enter a negative prompt',
137
+ value='lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry',
138
+ )
139
+ use_prompt_2 = gr.Checkbox(
140
+ label='Use prompt 2',
141
+ value=False
142
+ )
143
+ prompt_2 = gr.Text(
144
+ label='Prompt 2',
145
+ max_lines=1,
146
+ placeholder='Enter your prompt',
147
+ visible=False,
148
+ )
149
+ negative_prompt_2 = gr.Text(
150
+ label='Negative prompt 2',
151
+ max_lines=1,
152
+ placeholder='Enter a negative prompt',
153
+ visible=False,
154
+ )
155
+
156
+ # with gr.Row():
157
+ # aspect_ratio = gr.Dropdown(choices=choices, label="Aspect Ratio Preset", value=choices[2])
158
+ with gr.Row():
159
+ width = gr.Slider(
160
+ label='Width',
161
+ minimum=256,
162
+ maximum=MAX_IMAGE_SIZE,
163
+ step=32,
164
+ value=1024,
165
+ )
166
+ height = gr.Slider(
167
+ label='Height',
168
+ minimum=256,
169
+ maximum=MAX_IMAGE_SIZE,
170
+ step=32,
171
+ value=1024,
172
+ )
173
+ with gr.Accordion(label='Advanced Config', open=False):
174
+ with gr.Accordion(label='Conditioning Resolution', open=False):
175
+ with gr.Row():
176
+ original_width = gr.Slider(
177
+ label='Original Width',
178
+ minimum=1024,
179
+ maximum=4096,
180
+ step=32,
181
+ value=4096,
182
+ )
183
+ original_height = gr.Slider(
184
+ label='Original Height',
185
+ minimum=1024,
186
+ maximum=4096,
187
+ step=32,
188
+ value=4096,
189
+ )
190
+ with gr.Row():
191
+ target_width = gr.Slider(
192
+ label='Target Width',
193
+ minimum=1024,
194
+ maximum=4096,
195
+ step=32,
196
+ value=1024,
197
+ )
198
+ target_height = gr.Slider(
199
+ label='Target Height',
200
+ minimum=1024,
201
+ maximum=4096,
202
+ step=32,
203
+ value=1024,
204
+ )
205
+ seed = gr.Slider(label='Seed',
206
+ minimum=0,
207
+ maximum=MAX_SEED,
208
+ step=1,
209
+ value=0)
210
+
211
+ randomize_seed = gr.Checkbox(label='Randomize seed', value=True)
212
+ with gr.Row():
213
+ guidance_scale_base = gr.Slider(
214
+ label='Guidance scale',
215
+ minimum=1,
216
+ maximum=20,
217
+ step=0.1,
218
+ value=12.0)
219
+ num_inference_steps_base = gr.Slider(
220
+ label='Number of inference steps',
221
+ minimum=10,
222
+ maximum=100,
223
+ step=1,
224
+ value=50)
225
+
226
+ with gr.Column(scale=2):
227
+ with gr.Blocks():
228
+ run_button = gr.Button('Generate')
229
+ result = gr.Image(label='Result', show_label=False)
230
+
231
+ gr.Examples(examples=examples,
232
+ inputs=prompt,
233
+ outputs=result,
234
+ fn=generate,
235
+ cache_examples=CACHE_EXAMPLES)
236
+
237
+ use_prompt_2.change(
238
+ fn=lambda x: gr.update(visible=x),
239
+ inputs=use_prompt_2,
240
+ outputs=prompt_2,
241
+ queue=False,
242
+ api_name=False,
243
+ )
244
+ use_prompt_2.change(
245
+ fn=lambda x: gr.update(visible=x),
246
+ inputs=use_prompt_2,
247
+ outputs=negative_prompt_2,
248
+ queue=False,
249
+ api_name=False,
250
+ )
251
+
252
+ inputs = [
253
+ prompt,
254
+ negative_prompt,
255
+ prompt_2,
256
+ negative_prompt_2,
257
+ use_prompt_2,
258
+ seed,
259
+ width,
260
+ height,
261
+ target_width,
262
+ target_height,
263
+ original_width,
264
+ original_height,
265
+ guidance_scale_base,
266
+ num_inference_steps_base,
267
+ ]
268
+ prompt.submit(
269
+ fn=randomize_seed_fn,
270
+ inputs=[seed, randomize_seed],
271
+ outputs=seed,
272
+ queue=False,
273
+ api_name=False,
274
+ ).then(
275
+ fn=generate,
276
+ inputs=inputs,
277
+ outputs=result,
278
+ api_name='run',
279
+ )
280
+ negative_prompt.submit(
281
+ fn=randomize_seed_fn,
282
+ inputs=[seed, randomize_seed],
283
+ outputs=seed,
284
+ queue=False,
285
+ api_name=False,
286
+ ).then(
287
+ fn=generate,
288
+ inputs=inputs,
289
+ outputs=result,
290
+ api_name=False,
291
+ )
292
+ prompt_2.submit(
293
+ fn=randomize_seed_fn,
294
+ inputs=[seed, randomize_seed],
295
+ outputs=seed,
296
+ queue=False,
297
+ api_name=False,
298
+ ).then(
299
+ fn=generate,
300
+ inputs=inputs,
301
+ outputs=result,
302
+ api_name=False,
303
+ )
304
+ negative_prompt_2.submit(
305
+ fn=randomize_seed_fn,
306
+ inputs=[seed, randomize_seed],
307
+ outputs=seed,
308
+ queue=False,
309
+ api_name=False,
310
+ ).then(
311
+ fn=generate,
312
+ inputs=inputs,
313
+ outputs=result,
314
+ api_name=False,
315
+ )
316
+ run_button.click(
317
+ fn=randomize_seed_fn,
318
+ inputs=[seed, randomize_seed],
319
+ outputs=seed,
320
+ queue=False,
321
+ api_name=False,
322
+ ).then(
323
+ fn=generate,
324
+ inputs=inputs,
325
+ outputs=result,
326
+ api_name=False,
327
+ )
328
+
329
+ 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,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ h1 {
2
+ text-align: center;
3
+ font-size: 10vw; /* relative to the viewport width */
4
+ }
5
+
6
+ #duplicate-button {
7
+ margin: auto;
8
+ color: #fff;
9
+ background: #1565c0;
10
+ border-radius: 100vh;
11
+ }
12
+
13
+ #component-0 {
14
+ max-width: 80%; /* relative to the parent element's width */
15
+ margin: auto;
16
+ padding-top: 1.5rem;
17
+ }
18
+
19
+ /* You can also use media queries to adjust your style for different screen sizes */
20
+ @media (max-width: 600px) {
21
+ #component-0 {
22
+ max-width: 90%;
23
+ padding-top: 1rem;
24
+ }
25
+ }