dikdimon commited on
Commit
e2f163a
·
verified ·
1 Parent(s): 1e9f860

Upload sd-webui-semantic-guidance-sdxl-hybrid using SD-Hub

Browse files
.gitattributes CHANGED
@@ -191,3 +191,7 @@ negative_rejection_steering/Examples/skew_array.png filter=lfs diff=lfs merge=lf
191
  negative_rejection_steering/Examples/squash_matrix.png filter=lfs diff=lfs merge=lfs -text
192
  negative_rejection_steering/Examples/stretch_array.png filter=lfs diff=lfs merge=lfs -text
193
  negative_rejection_steering/icon.png filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
191
  negative_rejection_steering/Examples/squash_matrix.png filter=lfs diff=lfs merge=lfs -text
192
  negative_rejection_steering/Examples/stretch_array.png filter=lfs diff=lfs merge=lfs -text
193
  negative_rejection_steering/icon.png filter=lfs diff=lfs merge=lfs -text
194
+ sd-webui-semantic-guidance-sdxl-hybrid/samples/comparison.png filter=lfs diff=lfs merge=lfs -text
195
+ sd-webui-semantic-guidance-sdxl-hybrid/samples/concepts_1.jpg filter=lfs diff=lfs merge=lfs -text
196
+ sd-webui-semantic-guidance-sdxl-hybrid/samples/enhance.jpg filter=lfs diff=lfs merge=lfs -text
197
+ sd-webui-semantic-guidance-sdxl-hybrid/samples/style_change.jpg filter=lfs diff=lfs merge=lfs -text
sd-webui-semantic-guidance-sdxl-hybrid/LICENSE ADDED
@@ -0,0 +1,674 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ GNU GENERAL PUBLIC LICENSE
2
+ Version 3, 29 June 2007
3
+
4
+ Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
5
+ Everyone is permitted to copy and distribute verbatim copies
6
+ of this license document, but changing it is not allowed.
7
+
8
+ Preamble
9
+
10
+ The GNU General Public License is a free, copyleft license for
11
+ software and other kinds of works.
12
+
13
+ The licenses for most software and other practical works are designed
14
+ to take away your freedom to share and change the works. By contrast,
15
+ the GNU General Public License is intended to guarantee your freedom to
16
+ share and change all versions of a program--to make sure it remains free
17
+ software for all its users. We, the Free Software Foundation, use the
18
+ GNU General Public License for most of our software; it applies also to
19
+ any other work released this way by its authors. You can apply it to
20
+ your programs, too.
21
+
22
+ When we speak of free software, we are referring to freedom, not
23
+ price. Our General Public Licenses are designed to make sure that you
24
+ have the freedom to distribute copies of free software (and charge for
25
+ them if you wish), that you receive source code or can get it if you
26
+ want it, that you can change the software or use pieces of it in new
27
+ free programs, and that you know you can do these things.
28
+
29
+ To protect your rights, we need to prevent others from denying you
30
+ these rights or asking you to surrender the rights. Therefore, you have
31
+ certain responsibilities if you distribute copies of the software, or if
32
+ you modify it: responsibilities to respect the freedom of others.
33
+
34
+ For example, if you distribute copies of such a program, whether
35
+ gratis or for a fee, you must pass on to the recipients the same
36
+ freedoms that you received. You must make sure that they, too, receive
37
+ or can get the source code. And you must show them these terms so they
38
+ know their rights.
39
+
40
+ Developers that use the GNU GPL protect your rights with two steps:
41
+ (1) assert copyright on the software, and (2) offer you this License
42
+ giving you legal permission to copy, distribute and/or modify it.
43
+
44
+ For the developers' and authors' protection, the GPL clearly explains
45
+ that there is no warranty for this free software. For both users' and
46
+ authors' sake, the GPL requires that modified versions be marked as
47
+ changed, so that their problems will not be attributed erroneously to
48
+ authors of previous versions.
49
+
50
+ Some devices are designed to deny users access to install or run
51
+ modified versions of the software inside them, although the manufacturer
52
+ can do so. This is fundamentally incompatible with the aim of
53
+ protecting users' freedom to change the software. The systematic
54
+ pattern of such abuse occurs in the area of products for individuals to
55
+ use, which is precisely where it is most unacceptable. Therefore, we
56
+ have designed this version of the GPL to prohibit the practice for those
57
+ products. If such problems arise substantially in other domains, we
58
+ stand ready to extend this provision to those domains in future versions
59
+ of the GPL, as needed to protect the freedom of users.
60
+
61
+ Finally, every program is threatened constantly by software patents.
62
+ States should not allow patents to restrict development and use of
63
+ software on general-purpose computers, but in those that do, we wish to
64
+ avoid the special danger that patents applied to a free program could
65
+ make it effectively proprietary. To prevent this, the GPL assures that
66
+ patents cannot be used to render the program non-free.
67
+
68
+ The precise terms and conditions for copying, distribution and
69
+ modification follow.
70
+
71
+ TERMS AND CONDITIONS
72
+
73
+ 0. Definitions.
74
+
75
+ "This License" refers to version 3 of the GNU General Public License.
76
+
77
+ "Copyright" also means copyright-like laws that apply to other kinds of
78
+ works, such as semiconductor masks.
79
+
80
+ "The Program" refers to any copyrightable work licensed under this
81
+ License. Each licensee is addressed as "you". "Licensees" and
82
+ "recipients" may be individuals or organizations.
83
+
84
+ To "modify" a work means to copy from or adapt all or part of the work
85
+ in a fashion requiring copyright permission, other than the making of an
86
+ exact copy. The resulting work is called a "modified version" of the
87
+ earlier work or a work "based on" the earlier work.
88
+
89
+ A "covered work" means either the unmodified Program or a work based
90
+ on the Program.
91
+
92
+ To "propagate" a work means to do anything with it that, without
93
+ permission, would make you directly or secondarily liable for
94
+ infringement under applicable copyright law, except executing it on a
95
+ computer or modifying a private copy. Propagation includes copying,
96
+ distribution (with or without modification), making available to the
97
+ public, and in some countries other activities as well.
98
+
99
+ To "convey" a work means any kind of propagation that enables other
100
+ parties to make or receive copies. Mere interaction with a user through
101
+ a computer network, with no transfer of a copy, is not conveying.
102
+
103
+ An interactive user interface displays "Appropriate Legal Notices"
104
+ to the extent that it includes a convenient and prominently visible
105
+ feature that (1) displays an appropriate copyright notice, and (2)
106
+ tells the user that there is no warranty for the work (except to the
107
+ extent that warranties are provided), that licensees may convey the
108
+ work under this License, and how to view a copy of this License. If
109
+ the interface presents a list of user commands or options, such as a
110
+ menu, a prominent item in the list meets this criterion.
111
+
112
+ 1. Source Code.
113
+
114
+ The "source code" for a work means the preferred form of the work
115
+ for making modifications to it. "Object code" means any non-source
116
+ form of a work.
117
+
118
+ A "Standard Interface" means an interface that either is an official
119
+ standard defined by a recognized standards body, or, in the case of
120
+ interfaces specified for a particular programming language, one that
121
+ is widely used among developers working in that language.
122
+
123
+ The "System Libraries" of an executable work include anything, other
124
+ than the work as a whole, that (a) is included in the normal form of
125
+ packaging a Major Component, but which is not part of that Major
126
+ Component, and (b) serves only to enable use of the work with that
127
+ Major Component, or to implement a Standard Interface for which an
128
+ implementation is available to the public in source code form. A
129
+ "Major Component", in this context, means a major essential component
130
+ (kernel, window system, and so on) of the specific operating system
131
+ (if any) on which the executable work runs, or a compiler used to
132
+ produce the work, or an object code interpreter used to run it.
133
+
134
+ The "Corresponding Source" for a work in object code form means all
135
+ the source code needed to generate, install, and (for an executable
136
+ work) run the object code and to modify the work, including scripts to
137
+ control those activities. However, it does not include the work's
138
+ System Libraries, or general-purpose tools or generally available free
139
+ programs which are used unmodified in performing those activities but
140
+ which are not part of the work. For example, Corresponding Source
141
+ includes interface definition files associated with source files for
142
+ the work, and the source code for shared libraries and dynamically
143
+ linked subprograms that the work is specifically designed to require,
144
+ such as by intimate data communication or control flow between those
145
+ subprograms and other parts of the work.
146
+
147
+ The Corresponding Source need not include anything that users
148
+ can regenerate automatically from other parts of the Corresponding
149
+ Source.
150
+
151
+ The Corresponding Source for a work in source code form is that
152
+ same work.
153
+
154
+ 2. Basic Permissions.
155
+
156
+ All rights granted under this License are granted for the term of
157
+ copyright on the Program, and are irrevocable provided the stated
158
+ conditions are met. This License explicitly affirms your unlimited
159
+ permission to run the unmodified Program. The output from running a
160
+ covered work is covered by this License only if the output, given its
161
+ content, constitutes a covered work. This License acknowledges your
162
+ rights of fair use or other equivalent, as provided by copyright law.
163
+
164
+ You may make, run and propagate covered works that you do not
165
+ convey, without conditions so long as your license otherwise remains
166
+ in force. You may convey covered works to others for the sole purpose
167
+ of having them make modifications exclusively for you, or provide you
168
+ with facilities for running those works, provided that you comply with
169
+ the terms of this License in conveying all material for which you do
170
+ not control copyright. Those thus making or running the covered works
171
+ for you must do so exclusively on your behalf, under your direction
172
+ and control, on terms that prohibit them from making any copies of
173
+ your copyrighted material outside their relationship with you.
174
+
175
+ Conveying under any other circumstances is permitted solely under
176
+ the conditions stated below. Sublicensing is not allowed; section 10
177
+ makes it unnecessary.
178
+
179
+ 3. Protecting Users' Legal Rights From Anti-Circumvention Law.
180
+
181
+ No covered work shall be deemed part of an effective technological
182
+ measure under any applicable law fulfilling obligations under article
183
+ 11 of the WIPO copyright treaty adopted on 20 December 1996, or
184
+ similar laws prohibiting or restricting circumvention of such
185
+ measures.
186
+
187
+ When you convey a covered work, you waive any legal power to forbid
188
+ circumvention of technological measures to the extent such circumvention
189
+ is effected by exercising rights under this License with respect to
190
+ the covered work, and you disclaim any intention to limit operation or
191
+ modification of the work as a means of enforcing, against the work's
192
+ users, your or third parties' legal rights to forbid circumvention of
193
+ technological measures.
194
+
195
+ 4. Conveying Verbatim Copies.
196
+
197
+ You may convey verbatim copies of the Program's source code as you
198
+ receive it, in any medium, provided that you conspicuously and
199
+ appropriately publish on each copy an appropriate copyright notice;
200
+ keep intact all notices stating that this License and any
201
+ non-permissive terms added in accord with section 7 apply to the code;
202
+ keep intact all notices of the absence of any warranty; and give all
203
+ recipients a copy of this License along with the Program.
204
+
205
+ You may charge any price or no price for each copy that you convey,
206
+ and you may offer support or warranty protection for a fee.
207
+
208
+ 5. Conveying Modified Source Versions.
209
+
210
+ You may convey a work based on the Program, or the modifications to
211
+ produce it from the Program, in the form of source code under the
212
+ terms of section 4, provided that you also meet all of these conditions:
213
+
214
+ a) The work must carry prominent notices stating that you modified
215
+ it, and giving a relevant date.
216
+
217
+ b) The work must carry prominent notices stating that it is
218
+ released under this License and any conditions added under section
219
+ 7. This requirement modifies the requirement in section 4 to
220
+ "keep intact all notices".
221
+
222
+ c) You must license the entire work, as a whole, under this
223
+ License to anyone who comes into possession of a copy. This
224
+ License will therefore apply, along with any applicable section 7
225
+ additional terms, to the whole of the work, and all its parts,
226
+ regardless of how they are packaged. This License gives no
227
+ permission to license the work in any other way, but it does not
228
+ invalidate such permission if you have separately received it.
229
+
230
+ d) If the work has interactive user interfaces, each must display
231
+ Appropriate Legal Notices; however, if the Program has interactive
232
+ interfaces that do not display Appropriate Legal Notices, your
233
+ work need not make them do so.
234
+
235
+ A compilation of a covered work with other separate and independent
236
+ works, which are not by their nature extensions of the covered work,
237
+ and which are not combined with it such as to form a larger program,
238
+ in or on a volume of a storage or distribution medium, is called an
239
+ "aggregate" if the compilation and its resulting copyright are not
240
+ used to limit the access or legal rights of the compilation's users
241
+ beyond what the individual works permit. Inclusion of a covered work
242
+ in an aggregate does not cause this License to apply to the other
243
+ parts of the aggregate.
244
+
245
+ 6. Conveying Non-Source Forms.
246
+
247
+ You may convey a covered work in object code form under the terms
248
+ of sections 4 and 5, provided that you also convey the
249
+ machine-readable Corresponding Source under the terms of this License,
250
+ in one of these ways:
251
+
252
+ a) Convey the object code in, or embodied in, a physical product
253
+ (including a physical distribution medium), accompanied by the
254
+ Corresponding Source fixed on a durable physical medium
255
+ customarily used for software interchange.
256
+
257
+ b) Convey the object code in, or embodied in, a physical product
258
+ (including a physical distribution medium), accompanied by a
259
+ written offer, valid for at least three years and valid for as
260
+ long as you offer spare parts or customer support for that product
261
+ model, to give anyone who possesses the object code either (1) a
262
+ copy of the Corresponding Source for all the software in the
263
+ product that is covered by this License, on a durable physical
264
+ medium customarily used for software interchange, for a price no
265
+ more than your reasonable cost of physically performing this
266
+ conveying of source, or (2) access to copy the
267
+ Corresponding Source from a network server at no charge.
268
+
269
+ c) Convey individual copies of the object code with a copy of the
270
+ written offer to provide the Corresponding Source. This
271
+ alternative is allowed only occasionally and noncommercially, and
272
+ only if you received the object code with such an offer, in accord
273
+ with subsection 6b.
274
+
275
+ d) Convey the object code by offering access from a designated
276
+ place (gratis or for a charge), and offer equivalent access to the
277
+ Corresponding Source in the same way through the same place at no
278
+ further charge. You need not require recipients to copy the
279
+ Corresponding Source along with the object code. If the place to
280
+ copy the object code is a network server, the Corresponding Source
281
+ may be on a different server (operated by you or a third party)
282
+ that supports equivalent copying facilities, provided you maintain
283
+ clear directions next to the object code saying where to find the
284
+ Corresponding Source. Regardless of what server hosts the
285
+ Corresponding Source, you remain obligated to ensure that it is
286
+ available for as long as needed to satisfy these requirements.
287
+
288
+ e) Convey the object code using peer-to-peer transmission, provided
289
+ you inform other peers where the object code and Corresponding
290
+ Source of the work are being offered to the general public at no
291
+ charge under subsection 6d.
292
+
293
+ A separable portion of the object code, whose source code is excluded
294
+ from the Corresponding Source as a System Library, need not be
295
+ included in conveying the object code work.
296
+
297
+ A "User Product" is either (1) a "consumer product", which means any
298
+ tangible personal property which is normally used for personal, family,
299
+ or household purposes, or (2) anything designed or sold for incorporation
300
+ into a dwelling. In determining whether a product is a consumer product,
301
+ doubtful cases shall be resolved in favor of coverage. For a particular
302
+ product received by a particular user, "normally used" refers to a
303
+ typical or common use of that class of product, regardless of the status
304
+ of the particular user or of the way in which the particular user
305
+ actually uses, or expects or is expected to use, the product. A product
306
+ is a consumer product regardless of whether the product has substantial
307
+ commercial, industrial or non-consumer uses, unless such uses represent
308
+ the only significant mode of use of the product.
309
+
310
+ "Installation Information" for a User Product means any methods,
311
+ procedures, authorization keys, or other information required to install
312
+ and execute modified versions of a covered work in that User Product from
313
+ a modified version of its Corresponding Source. The information must
314
+ suffice to ensure that the continued functioning of the modified object
315
+ code is in no case prevented or interfered with solely because
316
+ modification has been made.
317
+
318
+ If you convey an object code work under this section in, or with, or
319
+ specifically for use in, a User Product, and the conveying occurs as
320
+ part of a transaction in which the right of possession and use of the
321
+ User Product is transferred to the recipient in perpetuity or for a
322
+ fixed term (regardless of how the transaction is characterized), the
323
+ Corresponding Source conveyed under this section must be accompanied
324
+ by the Installation Information. But this requirement does not apply
325
+ if neither you nor any third party retains the ability to install
326
+ modified object code on the User Product (for example, the work has
327
+ been installed in ROM).
328
+
329
+ The requirement to provide Installation Information does not include a
330
+ requirement to continue to provide support service, warranty, or updates
331
+ for a work that has been modified or installed by the recipient, or for
332
+ the User Product in which it has been modified or installed. Access to a
333
+ network may be denied when the modification itself materially and
334
+ adversely affects the operation of the network or violates the rules and
335
+ protocols for communication across the network.
336
+
337
+ Corresponding Source conveyed, and Installation Information provided,
338
+ in accord with this section must be in a format that is publicly
339
+ documented (and with an implementation available to the public in
340
+ source code form), and must require no special password or key for
341
+ unpacking, reading or copying.
342
+
343
+ 7. Additional Terms.
344
+
345
+ "Additional permissions" are terms that supplement the terms of this
346
+ License by making exceptions from one or more of its conditions.
347
+ Additional permissions that are applicable to the entire Program shall
348
+ be treated as though they were included in this License, to the extent
349
+ that they are valid under applicable law. If additional permissions
350
+ apply only to part of the Program, that part may be used separately
351
+ under those permissions, but the entire Program remains governed by
352
+ this License without regard to the additional permissions.
353
+
354
+ When you convey a copy of a covered work, you may at your option
355
+ remove any additional permissions from that copy, or from any part of
356
+ it. (Additional permissions may be written to require their own
357
+ removal in certain cases when you modify the work.) You may place
358
+ additional permissions on material, added by you to a covered work,
359
+ for which you have or can give appropriate copyright permission.
360
+
361
+ Notwithstanding any other provision of this License, for material you
362
+ add to a covered work, you may (if authorized by the copyright holders of
363
+ that material) supplement the terms of this License with terms:
364
+
365
+ a) Disclaiming warranty or limiting liability differently from the
366
+ terms of sections 15 and 16 of this License; or
367
+
368
+ b) Requiring preservation of specified reasonable legal notices or
369
+ author attributions in that material or in the Appropriate Legal
370
+ Notices displayed by works containing it; or
371
+
372
+ c) Prohibiting misrepresentation of the origin of that material, or
373
+ requiring that modified versions of such material be marked in
374
+ reasonable ways as different from the original version; or
375
+
376
+ d) Limiting the use for publicity purposes of names of licensors or
377
+ authors of the material; or
378
+
379
+ e) Declining to grant rights under trademark law for use of some
380
+ trade names, trademarks, or service marks; or
381
+
382
+ f) Requiring indemnification of licensors and authors of that
383
+ material by anyone who conveys the material (or modified versions of
384
+ it) with contractual assumptions of liability to the recipient, for
385
+ any liability that these contractual assumptions directly impose on
386
+ those licensors and authors.
387
+
388
+ All other non-permissive additional terms are considered "further
389
+ restrictions" within the meaning of section 10. If the Program as you
390
+ received it, or any part of it, contains a notice stating that it is
391
+ governed by this License along with a term that is a further
392
+ restriction, you may remove that term. If a license document contains
393
+ a further restriction but permits relicensing or conveying under this
394
+ License, you may add to a covered work material governed by the terms
395
+ of that license document, provided that the further restriction does
396
+ not survive such relicensing or conveying.
397
+
398
+ If you add terms to a covered work in accord with this section, you
399
+ must place, in the relevant source files, a statement of the
400
+ additional terms that apply to those files, or a notice indicating
401
+ where to find the applicable terms.
402
+
403
+ Additional terms, permissive or non-permissive, may be stated in the
404
+ form of a separately written license, or stated as exceptions;
405
+ the above requirements apply either way.
406
+
407
+ 8. Termination.
408
+
409
+ You may not propagate or modify a covered work except as expressly
410
+ provided under this License. Any attempt otherwise to propagate or
411
+ modify it is void, and will automatically terminate your rights under
412
+ this License (including any patent licenses granted under the third
413
+ paragraph of section 11).
414
+
415
+ However, if you cease all violation of this License, then your
416
+ license from a particular copyright holder is reinstated (a)
417
+ provisionally, unless and until the copyright holder explicitly and
418
+ finally terminates your license, and (b) permanently, if the copyright
419
+ holder fails to notify you of the violation by some reasonable means
420
+ prior to 60 days after the cessation.
421
+
422
+ Moreover, your license from a particular copyright holder is
423
+ reinstated permanently if the copyright holder notifies you of the
424
+ violation by some reasonable means, this is the first time you have
425
+ received notice of violation of this License (for any work) from that
426
+ copyright holder, and you cure the violation prior to 30 days after
427
+ your receipt of the notice.
428
+
429
+ Termination of your rights under this section does not terminate the
430
+ licenses of parties who have received copies or rights from you under
431
+ this License. If your rights have been terminated and not permanently
432
+ reinstated, you do not qualify to receive new licenses for the same
433
+ material under section 10.
434
+
435
+ 9. Acceptance Not Required for Having Copies.
436
+
437
+ You are not required to accept this License in order to receive or
438
+ run a copy of the Program. Ancillary propagation of a covered work
439
+ occurring solely as a consequence of using peer-to-peer transmission
440
+ to receive a copy likewise does not require acceptance. However,
441
+ nothing other than this License grants you permission to propagate or
442
+ modify any covered work. These actions infringe copyright if you do
443
+ not accept this License. Therefore, by modifying or propagating a
444
+ covered work, you indicate your acceptance of this License to do so.
445
+
446
+ 10. Automatic Licensing of Downstream Recipients.
447
+
448
+ Each time you convey a covered work, the recipient automatically
449
+ receives a license from the original licensors, to run, modify and
450
+ propagate that work, subject to this License. You are not responsible
451
+ for enforcing compliance by third parties with this License.
452
+
453
+ An "entity transaction" is a transaction transferring control of an
454
+ organization, or substantially all assets of one, or subdividing an
455
+ organization, or merging organizations. If propagation of a covered
456
+ work results from an entity transaction, each party to that
457
+ transaction who receives a copy of the work also receives whatever
458
+ licenses to the work the party's predecessor in interest had or could
459
+ give under the previous paragraph, plus a right to possession of the
460
+ Corresponding Source of the work from the predecessor in interest, if
461
+ the predecessor has it or can get it with reasonable efforts.
462
+
463
+ You may not impose any further restrictions on the exercise of the
464
+ rights granted or affirmed under this License. For example, you may
465
+ not impose a license fee, royalty, or other charge for exercise of
466
+ rights granted under this License, and you may not initiate litigation
467
+ (including a cross-claim or counterclaim in a lawsuit) alleging that
468
+ any patent claim is infringed by making, using, selling, offering for
469
+ sale, or importing the Program or any portion of it.
470
+
471
+ 11. Patents.
472
+
473
+ A "contributor" is a copyright holder who authorizes use under this
474
+ License of the Program or a work on which the Program is based. The
475
+ work thus licensed is called the contributor's "contributor version".
476
+
477
+ A contributor's "essential patent claims" are all patent claims
478
+ owned or controlled by the contributor, whether already acquired or
479
+ hereafter acquired, that would be infringed by some manner, permitted
480
+ by this License, of making, using, or selling its contributor version,
481
+ but do not include claims that would be infringed only as a
482
+ consequence of further modification of the contributor version. For
483
+ purposes of this definition, "control" includes the right to grant
484
+ patent sublicenses in a manner consistent with the requirements of
485
+ this License.
486
+
487
+ Each contributor grants you a non-exclusive, worldwide, royalty-free
488
+ patent license under the contributor's essential patent claims, to
489
+ make, use, sell, offer for sale, import and otherwise run, modify and
490
+ propagate the contents of its contributor version.
491
+
492
+ In the following three paragraphs, a "patent license" is any express
493
+ agreement or commitment, however denominated, not to enforce a patent
494
+ (such as an express permission to practice a patent or covenant not to
495
+ sue for patent infringement). To "grant" such a patent license to a
496
+ party means to make such an agreement or commitment not to enforce a
497
+ patent against the party.
498
+
499
+ If you convey a covered work, knowingly relying on a patent license,
500
+ and the Corresponding Source of the work is not available for anyone
501
+ to copy, free of charge and under the terms of this License, through a
502
+ publicly available network server or other readily accessible means,
503
+ then you must either (1) cause the Corresponding Source to be so
504
+ available, or (2) arrange to deprive yourself of the benefit of the
505
+ patent license for this particular work, or (3) arrange, in a manner
506
+ consistent with the requirements of this License, to extend the patent
507
+ license to downstream recipients. "Knowingly relying" means you have
508
+ actual knowledge that, but for the patent license, your conveying the
509
+ covered work in a country, or your recipient's use of the covered work
510
+ in a country, would infringe one or more identifiable patents in that
511
+ country that you have reason to believe are valid.
512
+
513
+ If, pursuant to or in connection with a single transaction or
514
+ arrangement, you convey, or propagate by procuring conveyance of, a
515
+ covered work, and grant a patent license to some of the parties
516
+ receiving the covered work authorizing them to use, propagate, modify
517
+ or convey a specific copy of the covered work, then the patent license
518
+ you grant is automatically extended to all recipients of the covered
519
+ work and works based on it.
520
+
521
+ A patent license is "discriminatory" if it does not include within
522
+ the scope of its coverage, prohibits the exercise of, or is
523
+ conditioned on the non-exercise of one or more of the rights that are
524
+ specifically granted under this License. You may not convey a covered
525
+ work if you are a party to an arrangement with a third party that is
526
+ in the business of distributing software, under which you make payment
527
+ to the third party based on the extent of your activity of conveying
528
+ the work, and under which the third party grants, to any of the
529
+ parties who would receive the covered work from you, a discriminatory
530
+ patent license (a) in connection with copies of the covered work
531
+ conveyed by you (or copies made from those copies), or (b) primarily
532
+ for and in connection with specific products or compilations that
533
+ contain the covered work, unless you entered into that arrangement,
534
+ or that patent license was granted, prior to 28 March 2007.
535
+
536
+ Nothing in this License shall be construed as excluding or limiting
537
+ any implied license or other defenses to infringement that may
538
+ otherwise be available to you under applicable patent law.
539
+
540
+ 12. No Surrender of Others' Freedom.
541
+
542
+ If conditions are imposed on you (whether by court order, agreement or
543
+ otherwise) that contradict the conditions of this License, they do not
544
+ excuse you from the conditions of this License. If you cannot convey a
545
+ covered work so as to satisfy simultaneously your obligations under this
546
+ License and any other pertinent obligations, then as a consequence you may
547
+ not convey it at all. For example, if you agree to terms that obligate you
548
+ to collect a royalty for further conveying from those to whom you convey
549
+ the Program, the only way you could satisfy both those terms and this
550
+ License would be to refrain entirely from conveying the Program.
551
+
552
+ 13. Use with the GNU Affero General Public License.
553
+
554
+ Notwithstanding any other provision of this License, you have
555
+ permission to link or combine any covered work with a work licensed
556
+ under version 3 of the GNU Affero General Public License into a single
557
+ combined work, and to convey the resulting work. The terms of this
558
+ License will continue to apply to the part which is the covered work,
559
+ but the special requirements of the GNU Affero General Public License,
560
+ section 13, concerning interaction through a network will apply to the
561
+ combination as such.
562
+
563
+ 14. Revised Versions of this License.
564
+
565
+ The Free Software Foundation may publish revised and/or new versions of
566
+ the GNU General Public License from time to time. Such new versions will
567
+ be similar in spirit to the present version, but may differ in detail to
568
+ address new problems or concerns.
569
+
570
+ Each version is given a distinguishing version number. If the
571
+ Program specifies that a certain numbered version of the GNU General
572
+ Public License "or any later version" applies to it, you have the
573
+ option of following the terms and conditions either of that numbered
574
+ version or of any later version published by the Free Software
575
+ Foundation. If the Program does not specify a version number of the
576
+ GNU General Public License, you may choose any version ever published
577
+ by the Free Software Foundation.
578
+
579
+ If the Program specifies that a proxy can decide which future
580
+ versions of the GNU General Public License can be used, that proxy's
581
+ public statement of acceptance of a version permanently authorizes you
582
+ to choose that version for the Program.
583
+
584
+ Later license versions may give you additional or different
585
+ permissions. However, no additional obligations are imposed on any
586
+ author or copyright holder as a result of your choosing to follow a
587
+ later version.
588
+
589
+ 15. Disclaimer of Warranty.
590
+
591
+ THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY
592
+ APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT
593
+ HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY
594
+ OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO,
595
+ THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
596
+ PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM
597
+ IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF
598
+ ALL NECESSARY SERVICING, REPAIR OR CORRECTION.
599
+
600
+ 16. Limitation of Liability.
601
+
602
+ IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
603
+ WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS
604
+ THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY
605
+ GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE
606
+ USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF
607
+ DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD
608
+ PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS),
609
+ EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF
610
+ SUCH DAMAGES.
611
+
612
+ 17. Interpretation of Sections 15 and 16.
613
+
614
+ If the disclaimer of warranty and limitation of liability provided
615
+ above cannot be given local legal effect according to their terms,
616
+ reviewing courts shall apply local law that most closely approximates
617
+ an absolute waiver of all civil liability in connection with the
618
+ Program, unless a warranty or assumption of liability accompanies a
619
+ copy of the Program in return for a fee.
620
+
621
+ END OF TERMS AND CONDITIONS
622
+
623
+ How to Apply These Terms to Your New Programs
624
+
625
+ If you develop a new program, and you want it to be of the greatest
626
+ possible use to the public, the best way to achieve this is to make it
627
+ free software which everyone can redistribute and change under these terms.
628
+
629
+ To do so, attach the following notices to the program. It is safest
630
+ to attach them to the start of each source file to most effectively
631
+ state the exclusion of warranty; and each file should have at least
632
+ the "copyright" line and a pointer to where the full notice is found.
633
+
634
+ <one line to give the program's name and a brief idea of what it does.>
635
+ Copyright (C) <year> <name of author>
636
+
637
+ This program is free software: you can redistribute it and/or modify
638
+ it under the terms of the GNU General Public License as published by
639
+ the Free Software Foundation, either version 3 of the License, or
640
+ (at your option) any later version.
641
+
642
+ This program is distributed in the hope that it will be useful,
643
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
644
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
645
+ GNU General Public License for more details.
646
+
647
+ You should have received a copy of the GNU General Public License
648
+ along with this program. If not, see <https://www.gnu.org/licenses/>.
649
+
650
+ Also add information on how to contact you by electronic and paper mail.
651
+
652
+ If the program does terminal interaction, make it output a short
653
+ notice like this when it starts in an interactive mode:
654
+
655
+ <program> Copyright (C) <year> <name of author>
656
+ This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
657
+ This is free software, and you are welcome to redistribute it
658
+ under certain conditions; type `show c' for details.
659
+
660
+ The hypothetical commands `show w' and `show c' should show the appropriate
661
+ parts of the General Public License. Of course, your program's commands
662
+ might be different; for a GUI interface, you would use an "about box".
663
+
664
+ You should also get your employer (if you work as a programmer) or school,
665
+ if any, to sign a "copyright disclaimer" for the program, if necessary.
666
+ For more information on this, and how to apply and follow the GNU GPL, see
667
+ <https://www.gnu.org/licenses/>.
668
+
669
+ The GNU General Public License does not permit incorporating your program
670
+ into proprietary programs. If your program is a subroutine library, you
671
+ may consider it more useful to permit linking proprietary applications with
672
+ the library. If this is what you want to do, use the GNU Lesser General
673
+ Public License instead of this License. But first, please read
674
+ <https://www.gnu.org/licenses/why-not-lgpl.html>.
sd-webui-semantic-guidance-sdxl-hybrid/README.md ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ## sd-webui-semantic-guidance
2
+ ### An unofficial implementation of *SEGA: Instructing Text-to-Image Models using Semantic Guidance* for Automatic1111 WebUI
3
+ Semantic Guidance gives you more control over the fine details of an image by emphasizing or de-emphasizing concepts given in an additional positive / negative text prompt.
4
+
5
+ ### Use Cases
6
+ - Altering the style of an image without changing the composition
7
+ ![image](samples/style_change.jpg)
8
+
9
+ - Adding concepts to an image
10
+ ![image](samples/concepts_1.jpg)
11
+
12
+ Please see the bottom of the page for more examples.
13
+
14
+ ### Issues / PR's are welcome!
15
+
16
+ ### Settings
17
+
18
+ * Positive / Negative Prompt: Concepts to emphasize / de-emphasize, separated by commas
19
+ * Warmup Steps: How many steps to wait before applying semantic guidance
20
+ * Edit Guidance Scale: Globally scale how much influence semantic guidance has on the image
21
+ * Tail Percentage Threshold: The percentage of latents to use when calculating the semantic guidance
22
+ * Momentum Scale: Scale the influence of the added momentum term
23
+ * Momentum Beta: Higher values will make the influence of the momentum term more stable
24
+
25
+ ### Feature / To-do List
26
+ - [x] SD XL support
27
+ - [x] Support A1111 prompt attention syntax and shortcuts for attention strength
28
+ - [x] Support restoring parameter values from infotext (Send to Txt2Img, Send to Img2Img, etc.)
29
+ - [x] XYZ plot support
30
+ - [x] SD 1.5 support (untested)
31
+ - [ ] Support semantic guidance prompt in regular Positive/Negative Prompt fields
32
+
33
+ ### Credits
34
+ - The authors of the original paper for their method (https://arxiv.org/abs/2301.12247):
35
+ ```
36
+ @misc{brack2023sega,
37
+ title={SEGA: Instructing Text-to-Image Models using Semantic Guidance},
38
+ author={Manuel Brack and Felix Friedrich and Dominik Hintersdorf and Lukas Struppek and Patrick Schramowski and Kristian Kersting},
39
+ year={2023},
40
+ eprint={2301.12247},
41
+ archivePrefix={arXiv},
42
+ primaryClass={cs.CV}
43
+ }
44
+ ```
45
+ - Official implementation of Semantic Guidance for Diffusion: (https://github.com/ml-research/semantic-image-editing)
46
+
47
+ - @udon-universe's extension templates: (https://github.com/udon-universe/stable-diffusion-webui-extension-templates)
48
+
49
+ ### More examples
50
+ - Enhancing the details of an image
51
+ ![image](samples/enhance.jpg)
52
+
53
+
sd-webui-semantic-guidance-sdxl-hybrid/samples/comparison.png ADDED

Git LFS Details

  • SHA256: 3198d50ba2a89143420bcaf1f9aa309a89768abf15d830968d1c06c1634ee4cb
  • Pointer size: 132 Bytes
  • Size of remote file: 2.17 MB
sd-webui-semantic-guidance-sdxl-hybrid/samples/concepts_1.jpg ADDED

Git LFS Details

  • SHA256: 8032d9370c3677c990beaf32c27e3b23d2f21dfb271362b48623c9f4456fab9b
  • Pointer size: 131 Bytes
  • Size of remote file: 436 kB
sd-webui-semantic-guidance-sdxl-hybrid/samples/enhance.jpg ADDED

Git LFS Details

  • SHA256: a74b774a244dccb05aba111341f8c52380e5ff9f6877012daf4fa67a52fb2a5f
  • Pointer size: 132 Bytes
  • Size of remote file: 1.1 MB
sd-webui-semantic-guidance-sdxl-hybrid/samples/style_change.jpg ADDED

Git LFS Details

  • SHA256: 74d47ae585b979e504786eed5f0739dcac20fc7396d34ba6b5f23cfeb120b2c4
  • Pointer size: 131 Bytes
  • Size of remote file: 572 kB
sd-webui-semantic-guidance-sdxl-hybrid/scripts/__pycache__/sega.cpython-310.pyc ADDED
Binary file (24.3 kB). View file
 
sd-webui-semantic-guidance-sdxl-hybrid/scripts/sega.py ADDED
@@ -0,0 +1,955 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import logging
2
+ import math
3
+ from os import environ
4
+
5
+ import gradio as gr
6
+ import scipy.stats as stats
7
+ import torch
8
+ import torch.nn.functional as F
9
+
10
+ import modules.scripts as scripts
11
+ from modules import prompt_parser, script_callbacks, shared
12
+ from modules.processing import StableDiffusionProcessing
13
+ from modules.prompt_parser import reconstruct_multicond_batch
14
+ from modules.script_callbacks import CFGDenoiserParams
15
+ from modules.sd_samplers_cfg_denoiser import pad_cond
16
+
17
+ logger = logging.getLogger(__name__)
18
+ logger.setLevel(environ.get("SD_WEBUI_LOG_LEVEL", logging.INFO))
19
+
20
+ """
21
+ An unofficial implementation of SEGA: Instructing Text-to-Image Models using Semantic Guidance for Automatic1111 WebUI.
22
+
23
+ This patched variant focuses on robust SDXL support and safer hybrid steering improvements:
24
+ - handles tensor and dict conditioning safely (SD 1.5 / SDXL)
25
+ - lets you steer cross-attention and pooled vector conditioning separately on SDXL
26
+ - applies SDXL vector edits with a safer warmup-aware schedule
27
+ - optionally adds T2I-Zero-inspired similarity correction and CTNMS-inspired token suppression in embedding space
28
+ - splits concept prompts on top-level commas only
29
+ """
30
+
31
+
32
+ class SegaStateParams:
33
+ def __init__(self):
34
+ self.concept_name = ""
35
+ self.v = {} # velocity
36
+ self.warmup_period: int = 10
37
+ self.edit_guidance_scale: float = 1.0
38
+ self.tail_percentage_threshold: float = 0.05
39
+ self.momentum_scale: float = 0.3
40
+ self.momentum_beta: float = 0.6
41
+ self.strength: float = 1.0
42
+ self.conditioning_mode: str = "Auto"
43
+ self.crossattn_scale: float = 1.0
44
+ self.vector_scale: float = 0.25
45
+ self.vector_schedule: str = "After warmup (soft)"
46
+ self.vector_ramp_steps: int = 8
47
+ self.similarity_correction_enabled: bool = False
48
+ self.similarity_topk_frac: float = 0.15
49
+ self.similarity_percentile: float = 0.75
50
+ self.similarity_window: int = 2
51
+ self.similarity_alpha: float = 0.20
52
+ self.embedding_suppression_enabled: bool = False
53
+ self.embedding_suppression_percentile: float = 0.80
54
+ self.embedding_suppression_alpha: float = 0.15
55
+
56
+
57
+ class SegaExtensionScript(scripts.Script):
58
+ def __init__(self):
59
+ self.cached_c = [None, None]
60
+
61
+ def title(self):
62
+ return "Semantic Guidance"
63
+
64
+ def show(self, is_img2img):
65
+ return scripts.AlwaysVisible
66
+
67
+ def ui(self, is_img2img):
68
+ with gr.Accordion("Semantic Guidance", open=False):
69
+ active = gr.Checkbox(value=False, default=False, label="Active", elem_id="sega_active")
70
+ with gr.Row():
71
+ prompt = gr.Textbox(lines=2, label="Prompt", elem_id="sega_prompt", elem_classes=["prompt"])
72
+ with gr.Row():
73
+ neg_prompt = gr.Textbox(lines=2, label="Negative Prompt", elem_id="sega_neg_prompt", elem_classes=["prompt"])
74
+ with gr.Row():
75
+ warmup = gr.Slider(
76
+ value=10,
77
+ minimum=0,
78
+ maximum=30,
79
+ step=1,
80
+ label="Warmup Period",
81
+ elem_id="sega_warmup",
82
+ info="How many steps to wait before applying semantic guidance.",
83
+ )
84
+ edit_guidance_scale = gr.Slider(
85
+ value=1.0,
86
+ minimum=0.0,
87
+ maximum=20.0,
88
+ step=0.01,
89
+ label="Edit Guidance Scale",
90
+ elem_id="sega_edit_guidance_scale",
91
+ info="Global scale of semantic guidance.",
92
+ )
93
+ tail_percentage_threshold = gr.Slider(
94
+ value=0.05,
95
+ minimum=0.0,
96
+ maximum=1.0,
97
+ step=0.01,
98
+ label="Tail Percentage Threshold",
99
+ elem_id="sega_tail_percentage_threshold",
100
+ info="Only edit values in the strongest tail of the semantic direction.",
101
+ )
102
+ momentum_scale = gr.Slider(
103
+ value=0.3,
104
+ minimum=0.0,
105
+ maximum=1.0,
106
+ step=0.01,
107
+ label="Momentum Scale",
108
+ elem_id="sega_momentum_scale",
109
+ info="Adds momentum to the semantic direction.",
110
+ )
111
+ momentum_beta = gr.Slider(
112
+ value=0.6,
113
+ minimum=0.0,
114
+ maximum=0.999,
115
+ step=0.01,
116
+ label="Momentum Beta",
117
+ elem_id="sega_momentum_beta",
118
+ info="Higher values make momentum more stable.",
119
+ )
120
+ with gr.Row():
121
+ conditioning_mode = gr.Dropdown(
122
+ choices=["Auto", "CrossAttn only", "CrossAttn + Vector"],
123
+ value="Auto",
124
+ label="SDXL Conditioning Mode",
125
+ elem_id="sega_conditioning_mode",
126
+ info="Auto = SD 1.5 uses cross-attention only; SDXL edits cross-attention and optionally pooled vector.",
127
+ )
128
+ crossattn_scale = gr.Slider(
129
+ value=1.0,
130
+ minimum=0.0,
131
+ maximum=2.0,
132
+ step=0.01,
133
+ label="CrossAttn Scale",
134
+ elem_id="sega_crossattn_scale",
135
+ info="Relative scale for token/cross-attention conditioning.",
136
+ )
137
+ vector_scale = gr.Slider(
138
+ value=0.25,
139
+ minimum=0.0,
140
+ maximum=2.0,
141
+ step=0.01,
142
+ label="Vector Scale",
143
+ elem_id="sega_vector_scale",
144
+ info="Relative scale for SDXL pooled vector conditioning. Lower is usually safer.",
145
+ )
146
+ with gr.Row():
147
+ vector_schedule = gr.Dropdown(
148
+ choices=["After warmup (soft)", "After warmup (step)", "Always on"],
149
+ value="After warmup (soft)",
150
+ label="Vector Schedule",
151
+ elem_id="sega_vector_schedule",
152
+ info="Safer SDXL vector steering: keep pooled vector off until warmup ends, then enable it softly or in one step.",
153
+ )
154
+ vector_ramp_steps = gr.Slider(
155
+ value=8,
156
+ minimum=0,
157
+ maximum=30,
158
+ step=1,
159
+ label="Vector Ramp Steps",
160
+ elem_id="sega_vector_ramp_steps",
161
+ info="Used by the soft vector schedule. 0 behaves like a step switch.",
162
+ )
163
+ with gr.Accordion("Hybrid Embedding Edits (experimental)", open=False):
164
+ with gr.Row():
165
+ similarity_correction_enabled = gr.Checkbox(
166
+ value=False,
167
+ label="Enable Similarity Correction",
168
+ elem_id="sega_similarity_correction_enabled",
169
+ info="T2I-Zero-inspired correction applied only to cross-attention embeddings before SEGA.",
170
+ )
171
+ embedding_suppression_enabled = gr.Checkbox(
172
+ value=False,
173
+ label="Enable Token Competition Suppression",
174
+ elem_id="sega_embedding_suppression_enabled",
175
+ info="CTNMS-inspired embedding-space suppression. Safer than attention hooks, but still experimental.",
176
+ )
177
+ with gr.Row():
178
+ similarity_topk_frac = gr.Slider(
179
+ value=0.15,
180
+ minimum=0.01,
181
+ maximum=0.50,
182
+ step=0.01,
183
+ label="Selected Token Fraction",
184
+ elem_id="sega_similarity_topk_frac",
185
+ info="Fraction of the most concept-salient tokens to target for hybrid edits.",
186
+ )
187
+ similarity_percentile = gr.Slider(
188
+ value=0.75,
189
+ minimum=0.10,
190
+ maximum=0.99,
191
+ step=0.01,
192
+ label="Similarity Percentile",
193
+ elem_id="sega_similarity_percentile",
194
+ info="Higher values make similarity correction more selective.",
195
+ )
196
+ similarity_window = gr.Slider(
197
+ value=2,
198
+ minimum=0,
199
+ maximum=12,
200
+ step=1,
201
+ label="Similarity Window",
202
+ elem_id="sega_similarity_window",
203
+ info="Local token window around each selected token for similarity correction.",
204
+ )
205
+ similarity_alpha = gr.Slider(
206
+ value=0.20,
207
+ minimum=0.0,
208
+ maximum=1.0,
209
+ step=0.01,
210
+ label="Similarity Alpha",
211
+ elem_id="sega_similarity_alpha",
212
+ info="Blend strength for similarity correction.",
213
+ )
214
+ with gr.Row():
215
+ embedding_suppression_percentile = gr.Slider(
216
+ value=0.80,
217
+ minimum=0.10,
218
+ maximum=0.99,
219
+ step=0.01,
220
+ label="Suppression Percentile",
221
+ elem_id="sega_embedding_suppression_percentile",
222
+ info="Only suppress strongly competing non-selected tokens above this percentile.",
223
+ )
224
+ embedding_suppression_alpha = gr.Slider(
225
+ value=0.15,
226
+ minimum=0.0,
227
+ maximum=1.0,
228
+ step=0.01,
229
+ label="Suppression Alpha",
230
+ elem_id="sega_embedding_suppression_alpha",
231
+ info="Strength of embedding-space token competition suppression.",
232
+ )
233
+
234
+ for comp in [
235
+ active,
236
+ prompt,
237
+ neg_prompt,
238
+ warmup,
239
+ edit_guidance_scale,
240
+ tail_percentage_threshold,
241
+ momentum_scale,
242
+ momentum_beta,
243
+ conditioning_mode,
244
+ crossattn_scale,
245
+ vector_scale,
246
+ vector_schedule,
247
+ vector_ramp_steps,
248
+ similarity_correction_enabled,
249
+ similarity_topk_frac,
250
+ similarity_percentile,
251
+ similarity_window,
252
+ similarity_alpha,
253
+ embedding_suppression_enabled,
254
+ embedding_suppression_percentile,
255
+ embedding_suppression_alpha,
256
+ ]:
257
+ comp.do_not_save_to_config = True
258
+
259
+ self.infotext_fields = [
260
+ (active, lambda d: gr.Checkbox.update(value="SEGA Active" in d)),
261
+ (prompt, "SEGA Prompt"),
262
+ (neg_prompt, "SEGA Negative Prompt"),
263
+ (warmup, "SEGA Warmup Period"),
264
+ (edit_guidance_scale, "SEGA Edit Guidance Scale"),
265
+ (tail_percentage_threshold, "SEGA Tail Percentage Threshold"),
266
+ (momentum_scale, "SEGA Momentum Scale"),
267
+ (momentum_beta, "SEGA Momentum Beta"),
268
+ (conditioning_mode, "SEGA SDXL Conditioning Mode"),
269
+ (crossattn_scale, "SEGA CrossAttn Scale"),
270
+ (vector_scale, "SEGA Vector Scale"),
271
+ (vector_schedule, "SEGA Vector Schedule"),
272
+ (vector_ramp_steps, "SEGA Vector Ramp Steps"),
273
+ (similarity_correction_enabled, lambda d: gr.Checkbox.update(value="SEGA Similarity Correction" in d)),
274
+ (similarity_topk_frac, "SEGA Selected Token Fraction"),
275
+ (similarity_percentile, "SEGA Similarity Percentile"),
276
+ (similarity_window, "SEGA Similarity Window"),
277
+ (similarity_alpha, "SEGA Similarity Alpha"),
278
+ (embedding_suppression_enabled, lambda d: gr.Checkbox.update(value="SEGA Token Competition Suppression" in d)),
279
+ (embedding_suppression_percentile, "SEGA Suppression Percentile"),
280
+ (embedding_suppression_alpha, "SEGA Suppression Alpha"),
281
+ ]
282
+ self.paste_field_names = [
283
+ "sega_active",
284
+ "sega_prompt",
285
+ "sega_neg_prompt",
286
+ "sega_warmup",
287
+ "sega_edit_guidance_scale",
288
+ "sega_tail_percentage_threshold",
289
+ "sega_momentum_scale",
290
+ "sega_momentum_beta",
291
+ "sega_conditioning_mode",
292
+ "sega_crossattn_scale",
293
+ "sega_vector_scale",
294
+ "sega_vector_schedule",
295
+ "sega_vector_ramp_steps",
296
+ "sega_similarity_correction_enabled",
297
+ "sega_similarity_topk_frac",
298
+ "sega_similarity_percentile",
299
+ "sega_similarity_window",
300
+ "sega_similarity_alpha",
301
+ "sega_embedding_suppression_enabled",
302
+ "sega_embedding_suppression_percentile",
303
+ "sega_embedding_suppression_alpha",
304
+ ]
305
+ return [
306
+ active,
307
+ prompt,
308
+ neg_prompt,
309
+ warmup,
310
+ edit_guidance_scale,
311
+ tail_percentage_threshold,
312
+ momentum_scale,
313
+ momentum_beta,
314
+ conditioning_mode,
315
+ crossattn_scale,
316
+ vector_scale,
317
+ vector_schedule,
318
+ vector_ramp_steps,
319
+ similarity_correction_enabled,
320
+ similarity_topk_frac,
321
+ similarity_percentile,
322
+ similarity_window,
323
+ similarity_alpha,
324
+ embedding_suppression_enabled,
325
+ embedding_suppression_percentile,
326
+ embedding_suppression_alpha,
327
+ ]
328
+
329
+ def process_batch(
330
+ self,
331
+ p: StableDiffusionProcessing,
332
+ active,
333
+ prompt,
334
+ neg_prompt,
335
+ warmup,
336
+ edit_guidance_scale,
337
+ tail_percentage_threshold,
338
+ momentum_scale,
339
+ momentum_beta,
340
+ conditioning_mode,
341
+ crossattn_scale,
342
+ vector_scale,
343
+ vector_schedule,
344
+ vector_ramp_steps,
345
+ similarity_correction_enabled,
346
+ similarity_topk_frac,
347
+ similarity_percentile,
348
+ similarity_window,
349
+ similarity_alpha,
350
+ embedding_suppression_enabled,
351
+ embedding_suppression_percentile,
352
+ embedding_suppression_alpha,
353
+ *args,
354
+ **kwargs,
355
+ ):
356
+ active = getattr(p, "sega_active", active)
357
+ if active is False:
358
+ return
359
+
360
+ prompt = getattr(p, "sega_prompt", prompt)
361
+ neg_prompt = getattr(p, "sega_neg_prompt", neg_prompt)
362
+ warmup = getattr(p, "sega_warmup", warmup)
363
+ edit_guidance_scale = getattr(p, "sega_edit_guidance_scale", edit_guidance_scale)
364
+ tail_percentage_threshold = getattr(p, "sega_tail_percentage_threshold", tail_percentage_threshold)
365
+ momentum_scale = getattr(p, "sega_momentum_scale", momentum_scale)
366
+ momentum_beta = getattr(p, "sega_momentum_beta", momentum_beta)
367
+ conditioning_mode = getattr(p, "sega_conditioning_mode", conditioning_mode)
368
+ crossattn_scale = getattr(p, "sega_crossattn_scale", crossattn_scale)
369
+ vector_scale = getattr(p, "sega_vector_scale", vector_scale)
370
+ vector_schedule = getattr(p, "sega_vector_schedule", vector_schedule)
371
+ vector_ramp_steps = getattr(p, "sega_vector_ramp_steps", vector_ramp_steps)
372
+ similarity_correction_enabled = getattr(p, "sega_similarity_correction_enabled", similarity_correction_enabled)
373
+ similarity_topk_frac = getattr(p, "sega_similarity_topk_frac", similarity_topk_frac)
374
+ similarity_percentile = getattr(p, "sega_similarity_percentile", similarity_percentile)
375
+ similarity_window = getattr(p, "sega_similarity_window", similarity_window)
376
+ similarity_alpha = getattr(p, "sega_similarity_alpha", similarity_alpha)
377
+ embedding_suppression_enabled = getattr(p, "sega_embedding_suppression_enabled", embedding_suppression_enabled)
378
+ embedding_suppression_percentile = getattr(p, "sega_embedding_suppression_percentile", embedding_suppression_percentile)
379
+ embedding_suppression_alpha = getattr(p, "sega_embedding_suppression_alpha", embedding_suppression_alpha)
380
+
381
+ p.extra_generation_params.update(
382
+ {
383
+ "SEGA Active": active,
384
+ "SEGA Prompt": prompt,
385
+ "SEGA Negative Prompt": neg_prompt,
386
+ "SEGA Warmup Period": warmup,
387
+ "SEGA Edit Guidance Scale": edit_guidance_scale,
388
+ "SEGA Tail Percentage Threshold": tail_percentage_threshold,
389
+ "SEGA Momentum Scale": momentum_scale,
390
+ "SEGA Momentum Beta": momentum_beta,
391
+ "SEGA SDXL Conditioning Mode": conditioning_mode,
392
+ "SEGA CrossAttn Scale": crossattn_scale,
393
+ "SEGA Vector Scale": vector_scale,
394
+ "SEGA Vector Schedule": vector_schedule,
395
+ "SEGA Vector Ramp Steps": vector_ramp_steps,
396
+ **({"SEGA Similarity Correction": True} if similarity_correction_enabled else {}),
397
+ "SEGA Selected Token Fraction": similarity_topk_frac,
398
+ "SEGA Similarity Percentile": similarity_percentile,
399
+ "SEGA Similarity Window": similarity_window,
400
+ "SEGA Similarity Alpha": similarity_alpha,
401
+ **({"SEGA Token Competition Suppression": True} if embedding_suppression_enabled else {}),
402
+ "SEGA Suppression Percentile": embedding_suppression_percentile,
403
+ "SEGA Suppression Alpha": embedding_suppression_alpha,
404
+ }
405
+ )
406
+
407
+ concept_prompts = self.parse_concept_prompt(prompt)
408
+ concept_prompts_neg = self.parse_concept_prompt(neg_prompt)
409
+ concept_prompts = [prompt_parser.parse_prompt_attention(concept)[0] for concept in concept_prompts]
410
+ concept_prompts_neg = [prompt_parser.parse_prompt_attention(neg_concept)[0] for neg_concept in concept_prompts_neg]
411
+ concept_prompts_neg = [[concept, -strength] for concept, strength in concept_prompts_neg]
412
+ concept_prompts.extend(concept_prompts_neg)
413
+
414
+ if not concept_prompts:
415
+ logger.warning("SEGA: active but no concept prompts were provided; skipping hook creation.")
416
+ return
417
+
418
+ concept_conds = []
419
+ for concept, strength in concept_prompts:
420
+ prompt_list = [concept] * p.batch_size
421
+ prompts = prompt_parser.SdConditioning(prompt_list, width=p.width, height=p.height)
422
+ c = p.get_conds_with_caching(
423
+ prompt_parser.get_multicond_learned_conditioning,
424
+ prompts,
425
+ p.steps,
426
+ [self.cached_c],
427
+ p.extra_network_data,
428
+ )
429
+ concept_conds.append([c, strength])
430
+
431
+ self.create_hook(
432
+ active,
433
+ concept_conds,
434
+ warmup,
435
+ edit_guidance_scale,
436
+ tail_percentage_threshold,
437
+ momentum_scale,
438
+ momentum_beta,
439
+ conditioning_mode,
440
+ crossattn_scale,
441
+ vector_scale,
442
+ vector_schedule,
443
+ vector_ramp_steps,
444
+ similarity_correction_enabled,
445
+ similarity_topk_frac,
446
+ similarity_percentile,
447
+ similarity_window,
448
+ similarity_alpha,
449
+ embedding_suppression_enabled,
450
+ embedding_suppression_percentile,
451
+ embedding_suppression_alpha,
452
+ )
453
+
454
+ def parse_concept_prompt(self, prompt: str) -> list[str]:
455
+ """Split on top-level commas only.
456
+
457
+ This keeps prompts like `(red, blue:1.2)` or `[a, b]` intact.
458
+ Backslash-escaped commas are also preserved.
459
+ """
460
+ if not prompt:
461
+ return []
462
+
463
+ parts = []
464
+ cur = []
465
+ depth_round = 0
466
+ depth_square = 0
467
+ depth_curly = 0
468
+ escape = False
469
+
470
+ for ch in prompt:
471
+ if escape:
472
+ cur.append(ch)
473
+ escape = False
474
+ continue
475
+ if ch == "\\":
476
+ cur.append(ch)
477
+ escape = True
478
+ continue
479
+ if ch == "(":
480
+ depth_round += 1
481
+ elif ch == ")" and depth_round > 0:
482
+ depth_round -= 1
483
+ elif ch == "[":
484
+ depth_square += 1
485
+ elif ch == "]" and depth_square > 0:
486
+ depth_square -= 1
487
+ elif ch == "{":
488
+ depth_curly += 1
489
+ elif ch == "}" and depth_curly > 0:
490
+ depth_curly -= 1
491
+
492
+ if ch == "," and depth_round == 0 and depth_square == 0 and depth_curly == 0:
493
+ item = "".join(cur).strip()
494
+ if item:
495
+ parts.append(item)
496
+ cur = []
497
+ continue
498
+
499
+ cur.append(ch)
500
+
501
+ item = "".join(cur).strip()
502
+ if item:
503
+ parts.append(item)
504
+ return parts
505
+
506
+ def create_hook(
507
+ self,
508
+ active,
509
+ concept_conds,
510
+ warmup,
511
+ edit_guidance_scale,
512
+ tail_percentage_threshold,
513
+ momentum_scale,
514
+ momentum_beta,
515
+ conditioning_mode,
516
+ crossattn_scale,
517
+ vector_scale,
518
+ vector_schedule,
519
+ vector_ramp_steps,
520
+ similarity_correction_enabled,
521
+ similarity_topk_frac,
522
+ similarity_percentile,
523
+ similarity_window,
524
+ similarity_alpha,
525
+ embedding_suppression_enabled,
526
+ embedding_suppression_percentile,
527
+ embedding_suppression_alpha,
528
+ ):
529
+ if not concept_conds:
530
+ return
531
+
532
+ concepts_sega_params = []
533
+ for _, strength in concept_conds:
534
+ sega_params = SegaStateParams()
535
+ sega_params.warmup_period = warmup
536
+ sega_params.edit_guidance_scale = edit_guidance_scale
537
+ sega_params.tail_percentage_threshold = tail_percentage_threshold
538
+ sega_params.momentum_scale = momentum_scale
539
+ sega_params.momentum_beta = momentum_beta
540
+ sega_params.strength = strength
541
+ sega_params.conditioning_mode = conditioning_mode
542
+ sega_params.crossattn_scale = crossattn_scale
543
+ sega_params.vector_scale = vector_scale
544
+ sega_params.vector_schedule = vector_schedule
545
+ sega_params.vector_ramp_steps = vector_ramp_steps
546
+ sega_params.similarity_correction_enabled = similarity_correction_enabled
547
+ sega_params.similarity_topk_frac = similarity_topk_frac
548
+ sega_params.similarity_percentile = similarity_percentile
549
+ sega_params.similarity_window = similarity_window
550
+ sega_params.similarity_alpha = similarity_alpha
551
+ sega_params.embedding_suppression_enabled = embedding_suppression_enabled
552
+ sega_params.embedding_suppression_percentile = embedding_suppression_percentile
553
+ sega_params.embedding_suppression_alpha = embedding_suppression_alpha
554
+ concepts_sega_params.append(sega_params)
555
+
556
+ callback = lambda params: self.on_cfg_denoiser_callback(params, concept_conds, concepts_sega_params)
557
+ logger.debug("Hooked callbacks")
558
+ script_callbacks.on_cfg_denoiser(callback)
559
+ script_callbacks.on_script_unloaded(self.unhook_callbacks)
560
+
561
+ def postprocess_batch(self, p, active, neg_text, *args, **kwargs):
562
+ active = getattr(p, "sega_active", active)
563
+ if active is False:
564
+ return
565
+ self.unhook_callbacks()
566
+
567
+ def unhook_callbacks(self):
568
+ logger.debug("Unhooked callbacks")
569
+ script_callbacks.remove_current_script_callbacks()
570
+
571
+ def normalize_conditioning(self, cond):
572
+ if isinstance(cond, dict):
573
+ return cond
574
+ return {"crossattn": cond}
575
+
576
+ def get_empty_crossattn(self):
577
+ return getattr(shared.sd_model, "cond_stage_model_empty_prompt", None)
578
+
579
+ def pad_crossattn_to_match(self, tensor: torch.Tensor, reference: torch.Tensor):
580
+ if tensor is None or reference is None:
581
+ return tensor
582
+ if tensor.dim() < 3 or reference.dim() < 3:
583
+ return tensor
584
+ if tensor.shape[1] == reference.shape[1]:
585
+ return tensor
586
+
587
+ empty = self.get_empty_crossattn()
588
+ if empty is None or empty.ndim < 2 or empty.shape[1] == 0:
589
+ logger.warning("SEGA: could not pad cross-attention conditioning safely; empty prompt tensor unavailable.")
590
+ return tensor
591
+
592
+ diff = reference.shape[1] - tensor.shape[1]
593
+ if diff <= 0:
594
+ return tensor
595
+ if diff % empty.shape[1] != 0:
596
+ logger.warning(
597
+ "SEGA: cross-attention length mismatch (%s vs %s) is not divisible by empty prompt width %s; leaving tensor unchanged.",
598
+ tensor.shape[1],
599
+ reference.shape[1],
600
+ empty.shape[1],
601
+ )
602
+ return tensor
603
+ return pad_cond(tensor, diff // empty.shape[1], empty)
604
+
605
+ def pad_conditioning_pair(self, cond_a: dict, cond_b: dict):
606
+ if "crossattn" not in cond_a or "crossattn" not in cond_b:
607
+ return cond_a, cond_b
608
+
609
+ a = cond_a["crossattn"]
610
+ b = cond_b["crossattn"]
611
+ if a.shape[1] == b.shape[1]:
612
+ return cond_a, cond_b
613
+
614
+ if a.shape[1] < b.shape[1]:
615
+ cond_a["crossattn"] = self.pad_crossattn_to_match(a, b)
616
+ else:
617
+ cond_b["crossattn"] = self.pad_crossattn_to_match(b, a)
618
+ return cond_a, cond_b
619
+
620
+ def get_vector_ramp_factor(self, sampling_step: int, sega_param: SegaStateParams) -> float:
621
+ if sega_param.conditioning_mode == "CrossAttn only":
622
+ return 0.0
623
+ schedule = getattr(sega_param, "vector_schedule", "After warmup (soft)")
624
+ if schedule == "Always on":
625
+ return 1.0
626
+
627
+ warmup = max(0, int(sega_param.warmup_period))
628
+ if sampling_step < warmup:
629
+ return 0.0
630
+
631
+ ramp_steps = max(0, int(getattr(sega_param, "vector_ramp_steps", 8)))
632
+ if schedule == "After warmup (step)" or ramp_steps == 0:
633
+ return 1.0
634
+
635
+ progress = min(1.0, max(0.0, (sampling_step - warmup + 1) / max(1, ramp_steps)))
636
+ return progress * progress * (3.0 - 2.0 * progress)
637
+
638
+ def get_key_scale(self, key: str, sega_param: SegaStateParams, sampling_step: int) -> float:
639
+ if key == "crossattn":
640
+ return sega_param.crossattn_scale
641
+ if key == "vector":
642
+ if sega_param.conditioning_mode == "CrossAttn only":
643
+ return 0.0
644
+ return sega_param.vector_scale * self.get_vector_ramp_factor(sampling_step, sega_param)
645
+ return 1.0 if sega_param.conditioning_mode == "CrossAttn + Vector" else 0.0
646
+
647
+ def get_empty_reference_for(self, tensor: torch.Tensor):
648
+ if tensor is None or tensor.dim() < 3:
649
+ return None
650
+
651
+ empty = self.get_empty_crossattn()
652
+ if empty is None:
653
+ return None
654
+ if empty.dim() == 2:
655
+ empty = empty.unsqueeze(0)
656
+
657
+ empty = empty.to(dtype=tensor.dtype, device=tensor.device)
658
+ if empty.shape[0] != tensor.shape[0]:
659
+ if empty.shape[0] == 1:
660
+ empty = empty.repeat(tensor.shape[0], 1, 1)
661
+ else:
662
+ empty = empty[:1].repeat(tensor.shape[0], 1, 1)
663
+
664
+ if empty.shape[1] != tensor.shape[1]:
665
+ empty = self.pad_crossattn_to_match(empty, tensor)
666
+ if empty is None or empty.shape[1] != tensor.shape[1]:
667
+ return None
668
+ return empty
669
+
670
+ def select_salient_token_indices(self, f: torch.Tensor, reference: torch.Tensor | None, topk_frac: float, max_tokens: int = 8) -> list[int]:
671
+ if f.ndim != 2 or f.shape[0] == 0:
672
+ return []
673
+
674
+ if reference is not None and reference.shape == f.shape:
675
+ scores = torch.linalg.norm((f - reference).float(), dim=-1)
676
+ else:
677
+ scores = torch.linalg.norm(f.float(), dim=-1)
678
+
679
+ k = max(1, int(math.ceil(float(f.shape[0]) * max(0.01, topk_frac))))
680
+ k = min(int(f.shape[0]), max_tokens, k)
681
+ vals, idx = torch.topk(scores, k=k)
682
+ idx = idx[torch.isfinite(vals)]
683
+ if idx.numel() == 0:
684
+ return []
685
+ return idx.sort().values.tolist()
686
+
687
+ def correction_by_similarities(self, f: torch.Tensor, selected_tokens: list[int], percentile: float, window: int, alpha: float) -> torch.Tensor:
688
+ if alpha <= 0.0 or not selected_tokens or f.ndim != 2:
689
+ return f
690
+
691
+ corrected = f.detach().clone()
692
+ normed = F.normalize(f.float(), dim=-1, eps=1e-8)
693
+ n = f.shape[0]
694
+ percentile = min(0.999, max(0.001, float(percentile)))
695
+ window = max(0, int(window))
696
+
697
+ for token_idx in selected_tokens:
698
+ if token_idx < 0 or token_idx >= n:
699
+ continue
700
+ scores = torch.mv(normed, normed[token_idx])
701
+ positive = scores[scores > 0]
702
+ if positive.numel() == 0:
703
+ continue
704
+ tau = torch.quantile(positive, percentile)
705
+ weights = torch.where(scores > tau, scores, torch.zeros_like(scores))
706
+ if window > 0:
707
+ mask = torch.zeros_like(weights)
708
+ start = max(0, token_idx - window)
709
+ end = min(n, token_idx + window + 1)
710
+ mask[start:end] = 1
711
+ weights = weights * mask
712
+ total = weights.sum()
713
+ if total <= 0:
714
+ continue
715
+ weights = (weights / total).to(dtype=f.dtype, device=f.device)
716
+ mixed = torch.sum(weights.unsqueeze(1) * f, dim=0)
717
+ corrected[token_idx] = ((1.0 - alpha) * f[token_idx]) + (alpha * mixed)
718
+ return corrected
719
+
720
+ def suppress_competing_tokens(self, f: torch.Tensor, selected_tokens: list[int], percentile: float, alpha: float) -> torch.Tensor:
721
+ if alpha <= 0.0 or not selected_tokens or f.ndim != 2:
722
+ return f
723
+
724
+ corrected = f.detach().clone()
725
+ normed = F.normalize(corrected.float(), dim=-1, eps=1e-8)
726
+ selected = normed[selected_tokens]
727
+ similarities = torch.matmul(normed, selected.transpose(0, 1))
728
+ max_sim, _ = similarities.max(dim=1)
729
+
730
+ suppression_scores = max_sim.clone()
731
+ suppression_scores[selected_tokens] = -1.0
732
+ positive = suppression_scores[suppression_scores > 0]
733
+ if positive.numel() == 0:
734
+ return corrected
735
+
736
+ percentile = min(0.999, max(0.001, float(percentile)))
737
+ tau = torch.quantile(positive, percentile)
738
+ mask = max_sim > tau
739
+ if selected_tokens:
740
+ mask[torch.tensor(selected_tokens, device=mask.device)] = False
741
+ if not torch.any(mask):
742
+ return corrected
743
+
744
+ denom = max(1e-6, float(1.0 - tau))
745
+ damp = 1.0 - (alpha * ((max_sim - tau).clamp(min=0.0) / denom).clamp(max=1.0))
746
+ corrected[mask] = corrected[mask] * damp[mask].unsqueeze(1).to(dtype=corrected.dtype)
747
+ return corrected
748
+
749
+ def apply_hybrid_crossattn_edits(self, tensor: torch.Tensor, sega_param: SegaStateParams) -> torch.Tensor:
750
+ if tensor is None or tensor.dim() != 3:
751
+ return tensor
752
+ if not getattr(sega_param, "similarity_correction_enabled", False) and not getattr(sega_param, "embedding_suppression_enabled", False):
753
+ return tensor
754
+
755
+ reference = self.get_empty_reference_for(tensor)
756
+ edited = tensor.detach().clone()
757
+ for batch_idx in range(tensor.shape[0]):
758
+ ref = None if reference is None else reference[batch_idx]
759
+ selected_tokens = self.select_salient_token_indices(
760
+ edited[batch_idx],
761
+ ref,
762
+ getattr(sega_param, "similarity_topk_frac", 0.15),
763
+ )
764
+ if getattr(sega_param, "similarity_correction_enabled", False):
765
+ edited[batch_idx] = self.correction_by_similarities(
766
+ edited[batch_idx],
767
+ selected_tokens,
768
+ getattr(sega_param, "similarity_percentile", 0.75),
769
+ getattr(sega_param, "similarity_window", 2),
770
+ getattr(sega_param, "similarity_alpha", 0.20),
771
+ )
772
+ if getattr(sega_param, "embedding_suppression_enabled", False):
773
+ edited[batch_idx] = self.suppress_competing_tokens(
774
+ edited[batch_idx],
775
+ selected_tokens,
776
+ getattr(sega_param, "embedding_suppression_percentile", 0.80),
777
+ getattr(sega_param, "embedding_suppression_alpha", 0.15),
778
+ )
779
+ return edited
780
+
781
+ def on_cfg_denoiser_callback(self, params: CFGDenoiserParams, concept_conds, sega_params: list[SegaStateParams]):
782
+ if not concept_conds or not sega_params:
783
+ return
784
+
785
+ sampling_step = params.sampling_step
786
+ text_cond_is_dict = isinstance(params.text_cond, dict)
787
+ text_uncond_is_dict = isinstance(params.text_uncond, dict)
788
+ text_cond = self.normalize_conditioning(params.text_cond)
789
+ text_uncond = self.normalize_conditioning(params.text_uncond)
790
+ text_cond, text_uncond = self.pad_conditioning_pair(text_cond, text_uncond)
791
+ if not text_cond_is_dict:
792
+ params.text_cond = text_cond["crossattn"]
793
+ if not text_uncond_is_dict:
794
+ params.text_uncond = text_uncond["crossattn"]
795
+
796
+ batch_conds_list = []
797
+ batch_tensor = {}
798
+
799
+ for idx, (concept_cond, _) in enumerate(concept_conds):
800
+ conds_list, tensor_dict = reconstruct_multicond_batch(concept_cond, sampling_step)
801
+ tensor_dict = self.normalize_conditioning(tensor_dict)
802
+ tensor_dict, _ = self.pad_conditioning_pair(tensor_dict, text_uncond)
803
+ if "crossattn" in tensor_dict:
804
+ tensor_dict["crossattn"] = self.apply_hybrid_crossattn_edits(tensor_dict["crossattn"], sega_params[idx])
805
+
806
+ for key, tensor in tensor_dict.items():
807
+ tensor = tensor.unsqueeze(0)
808
+ if key not in batch_tensor:
809
+ batch_tensor[key] = tensor
810
+ else:
811
+ batch_tensor[key] = torch.cat((batch_tensor[key], tensor), dim=0)
812
+ batch_conds_list.append(conds_list)
813
+
814
+ self.sega_routine_batch(params, batch_conds_list, batch_tensor, sega_params, text_cond, text_uncond)
815
+
816
+ def make_tuple_dim(self, value):
817
+ if isinstance(value, torch.Tensor):
818
+ value = value.dim()
819
+ return (-1,) + (1,) * (value - 1)
820
+
821
+ def sega_routine_batch(
822
+ self,
823
+ params: CFGDenoiserParams,
824
+ batch_conds_list,
825
+ batch_tensor,
826
+ sega_params: list[SegaStateParams],
827
+ text_cond,
828
+ text_uncond,
829
+ ):
830
+ warmup_period = sega_params[0].warmup_period
831
+ edit_guidance_scale = sega_params[0].edit_guidance_scale
832
+ tail_percentage_threshold = sega_params[0].tail_percentage_threshold
833
+ momentum_scale = sega_params[0].momentum_scale
834
+ momentum_beta = sega_params[0].momentum_beta
835
+ sampling_step = params.sampling_step
836
+
837
+ edit_dir_dict = {}
838
+
839
+ for key, concept_cond in batch_tensor.items():
840
+ key_scale = self.get_key_scale(key, sega_params[0], sampling_step)
841
+ if key_scale == 0.0:
842
+ continue
843
+
844
+ strength = torch.tensor([sp.strength for sp in sega_params], dtype=concept_cond.dtype, device=concept_cond.device)
845
+ strength = strength.view(self.make_tuple_dim(concept_cond))
846
+
847
+ if key not in edit_dir_dict:
848
+ edit_dir_dict[key] = torch.zeros_like(concept_cond, dtype=concept_cond.dtype, device=concept_cond.device)
849
+
850
+ inside_dim = tuple(range(-concept_cond.dim() + 1, 0))
851
+ cond_mean = torch.mean(concept_cond, dim=inside_dim)
852
+ cond_std = torch.std(concept_cond, dim=inside_dim)
853
+
854
+ edit_dir = concept_cond - text_uncond[key]
855
+ edit_dir = torch.mul(strength, edit_dir)
856
+
857
+ tail_q = min(max(1e-6, 1.0 - float(tail_percentage_threshold)), 1.0 - 1e-6)
858
+ upper_z = stats.norm.ppf(tail_q)
859
+ upper_threshold = cond_mean + (upper_z * cond_std)
860
+ upper_threshold_reshaped = upper_threshold.view(self.make_tuple_dim(concept_cond))
861
+
862
+ zero_tensor = torch.zeros_like(concept_cond, dtype=concept_cond.dtype, device=concept_cond.device)
863
+ effective_scale = edit_guidance_scale * key_scale
864
+ scale_tensor = torch.ones_like(concept_cond, dtype=concept_cond.dtype, device=concept_cond.device) * effective_scale
865
+ edit_dir_abs = edit_dir.abs()
866
+ scale_tensor = torch.where(edit_dir_abs > upper_threshold_reshaped, scale_tensor, zero_tensor)
867
+
868
+ guidance_strength = 0.0 if sampling_step < warmup_period else 1.0
869
+ edit_dir = torch.mul(scale_tensor, edit_dir)
870
+ edit_dir_dict[key] = edit_dir_dict[key] + guidance_strength * edit_dir
871
+
872
+ for i, sega_param in enumerate(sega_params):
873
+ for key, direction in edit_dir_dict.items():
874
+ if key not in sega_param.v:
875
+ slice_idx = 1 - direction.dim()
876
+ sega_param.v[key] = torch.zeros(direction.shape[slice_idx:], dtype=direction.dtype, device=direction.device)
877
+
878
+ v_t = sega_param.v[key]
879
+ direction[i] = direction[i] + torch.mul(momentum_scale, v_t)
880
+ v_t_1 = momentum_beta * ((1 - momentum_beta) * v_t) * direction[i]
881
+
882
+ if sampling_step >= warmup_period:
883
+ if isinstance(params.text_cond, dict):
884
+ if key in params.text_cond:
885
+ params.text_cond[key] = params.text_cond[key] + direction[i]
886
+ else:
887
+ if key == "crossattn":
888
+ params.text_cond = params.text_cond + direction[i]
889
+
890
+ sega_param.v[key] = v_t_1
891
+
892
+
893
+ # XYZ Plot
894
+ # Based on @mcmonkey4eva's XYZ Plot implementation here:
895
+ # https://github.com/mcmonkeyprojects/sd-dynamic-thresholding/blob/master/scripts/dynamic_thresholding.py
896
+
897
+ def sega_apply_override(field, boolean: bool = False, activate: bool = False):
898
+ def fun(p, x, xs):
899
+ if boolean:
900
+ x = True if x.lower() == "true" else False
901
+ if activate and not hasattr(p, "sega_active"):
902
+ setattr(p, "sega_active", True)
903
+ setattr(p, field, x)
904
+ return fun
905
+
906
+
907
+
908
+ def sega_apply_field(field):
909
+ def fun(p, x, xs):
910
+ if not hasattr(p, "sega_active"):
911
+ setattr(p, "sega_active", True)
912
+ setattr(p, field, x)
913
+
914
+ return fun
915
+
916
+
917
+
918
+ def make_axis_options():
919
+ xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ in ("xyz_grid.py", "scripts.xyz_grid")][0].module
920
+ extra_axis_options = {
921
+ xyz_grid.AxisOption("[Semantic Guidance] Active", str, sega_apply_override("sega_active", boolean=True), choices=xyz_grid.boolean_choice(reverse=True)),
922
+ xyz_grid.AxisOption("[Semantic Guidance] Prompt", str, sega_apply_field("sega_prompt")),
923
+ xyz_grid.AxisOption("[Semantic Guidance] Negative Prompt", str, sega_apply_field("sega_neg_prompt")),
924
+ xyz_grid.AxisOption("[Semantic Guidance] Warmup Steps", int, sega_apply_field("sega_warmup")),
925
+ xyz_grid.AxisOption("[Semantic Guidance] Guidance Scale", float, sega_apply_field("sega_edit_guidance_scale")),
926
+ xyz_grid.AxisOption("[Semantic Guidance] Tail Percentage Threshold", float, sega_apply_field("sega_tail_percentage_threshold")),
927
+ xyz_grid.AxisOption("[Semantic Guidance] Momentum Scale", float, sega_apply_field("sega_momentum_scale")),
928
+ xyz_grid.AxisOption("[Semantic Guidance] Momentum Beta", float, sega_apply_field("sega_momentum_beta")),
929
+ xyz_grid.AxisOption("[Semantic Guidance] SDXL Conditioning Mode", str, sega_apply_field("sega_conditioning_mode")),
930
+ xyz_grid.AxisOption("[Semantic Guidance] CrossAttn Scale", float, sega_apply_field("sega_crossattn_scale")),
931
+ xyz_grid.AxisOption("[Semantic Guidance] Vector Scale", float, sega_apply_field("sega_vector_scale")),
932
+ xyz_grid.AxisOption("[Semantic Guidance] Vector Schedule", str, sega_apply_field("sega_vector_schedule")),
933
+ xyz_grid.AxisOption("[Semantic Guidance] Vector Ramp Steps", int, sega_apply_field("sega_vector_ramp_steps")),
934
+ xyz_grid.AxisOption("[Semantic Guidance] Similarity Correction", str, sega_apply_override("sega_similarity_correction_enabled", boolean=True, activate=True), choices=xyz_grid.boolean_choice(reverse=True)),
935
+ xyz_grid.AxisOption("[Semantic Guidance] Selected Token Fraction", float, sega_apply_field("sega_similarity_topk_frac")),
936
+ xyz_grid.AxisOption("[Semantic Guidance] Similarity Percentile", float, sega_apply_field("sega_similarity_percentile")),
937
+ xyz_grid.AxisOption("[Semantic Guidance] Similarity Window", int, sega_apply_field("sega_similarity_window")),
938
+ xyz_grid.AxisOption("[Semantic Guidance] Similarity Alpha", float, sega_apply_field("sega_similarity_alpha")),
939
+ xyz_grid.AxisOption("[Semantic Guidance] Token Competition Suppression", str, sega_apply_override("sega_embedding_suppression_enabled", boolean=True, activate=True), choices=xyz_grid.boolean_choice(reverse=True)),
940
+ xyz_grid.AxisOption("[Semantic Guidance] Suppression Percentile", float, sega_apply_field("sega_embedding_suppression_percentile")),
941
+ xyz_grid.AxisOption("[Semantic Guidance] Suppression Alpha", float, sega_apply_field("sega_embedding_suppression_alpha")),
942
+ }
943
+ if not any("[Semantic Guidance]" in x.label for x in xyz_grid.axis_options):
944
+ xyz_grid.axis_options.extend(extra_axis_options)
945
+
946
+
947
+
948
+ def callback_before_ui():
949
+ try:
950
+ make_axis_options()
951
+ except Exception:
952
+ logger.exception("Semantic Guidance: Error while making axis options")
953
+
954
+
955
+ script_callbacks.on_before_ui(callback_before_ui)