Upload 7 files
Browse files- sd-webui-semantic-guidance-sdxl-hybrid/LICENSE +674 -0
- sd-webui-semantic-guidance-sdxl-hybrid/README.md +53 -0
- sd-webui-semantic-guidance-sdxl-hybrid/samples/comparison.png +3 -0
- sd-webui-semantic-guidance-sdxl-hybrid/samples/concepts_1.jpg +3 -0
- sd-webui-semantic-guidance-sdxl-hybrid/samples/enhance.jpg +3 -0
- sd-webui-semantic-guidance-sdxl-hybrid/samples/style_change.jpg +3 -0
- sd-webui-semantic-guidance-sdxl-hybrid/scripts/sega.py +1030 -0
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 |
+

|
| 8 |
+
|
| 9 |
+
- Adding concepts to an image
|
| 10 |
+

|
| 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 |
+

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