Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
S
Stable Diffusion Webui
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Stable Diffusion Webui
Commits
2cb3b0be
Commit
2cb3b0be
authored
May 07, 2023
by
AUTOMATIC
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
if present, use Lora's "ss_output_name" field to refer to it in prompt
parent
335428c2
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
12 additions
and
4 deletions
+12
-4
extensions-builtin/Lora/extra_networks_lora.py
extensions-builtin/Lora/extra_networks_lora.py
+1
-0
extensions-builtin/Lora/lora.py
extensions-builtin/Lora/lora.py
+10
-3
extensions-builtin/Lora/ui_extra_networks_lora.py
extensions-builtin/Lora/ui_extra_networks_lora.py
+1
-1
No files found.
extensions-builtin/Lora/extra_networks_lora.py
View file @
2cb3b0be
from
modules
import
extra_networks
,
shared
import
lora
class
ExtraNetworkLora
(
extra_networks
.
ExtraNetwork
):
def
__init__
(
self
):
super
()
.
__init__
(
'lora'
)
...
...
extensions-builtin/Lora/lora.py
View file @
2cb3b0be
...
...
@@ -93,6 +93,7 @@ class LoraOnDisk:
self
.
metadata
=
m
self
.
ssmd_cover_images
=
self
.
metadata
.
pop
(
'ssmd_cover_images'
,
None
)
# those are cover images and they are too big to display in UI as text
self
.
alias
=
self
.
metadata
.
get
(
'ss_output_name'
,
self
.
name
)
class
LoraModule
:
...
...
@@ -199,11 +200,11 @@ def load_loras(names, multipliers=None):
loaded_loras
.
clear
()
loras_on_disk
=
[
available_loras
.
get
(
name
,
None
)
for
name
in
names
]
loras_on_disk
=
[
available_lora
_aliase
s
.
get
(
name
,
None
)
for
name
in
names
]
if
any
([
x
is
None
for
x
in
loras_on_disk
]):
list_available_loras
()
loras_on_disk
=
[
available_loras
.
get
(
name
,
None
)
for
name
in
names
]
loras_on_disk
=
[
available_lora
_aliase
s
.
get
(
name
,
None
)
for
name
in
names
]
for
i
,
name
in
enumerate
(
names
):
lora
=
already_loaded
.
get
(
name
,
None
)
...
...
@@ -343,6 +344,7 @@ def lora_MultiheadAttention_load_state_dict(self, *args, **kwargs):
def
list_available_loras
():
available_loras
.
clear
()
available_lora_aliases
.
clear
()
os
.
makedirs
(
shared
.
cmd_opts
.
lora_dir
,
exist_ok
=
True
)
...
...
@@ -356,11 +358,16 @@ def list_available_loras():
continue
name
=
os
.
path
.
splitext
(
os
.
path
.
basename
(
filename
))[
0
]
entry
=
LoraOnDisk
(
name
,
filename
)
available_loras
[
name
]
=
LoraOnDisk
(
name
,
filename
)
available_loras
[
name
]
=
entry
available_lora_aliases
[
name
]
=
entry
available_lora_aliases
[
entry
.
alias
]
=
entry
available_loras
=
{}
available_lora_aliases
=
{}
loaded_loras
=
[]
list_available_loras
()
extensions-builtin/Lora/ui_extra_networks_lora.py
View file @
2cb3b0be
...
...
@@ -21,7 +21,7 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
"preview"
:
self
.
find_preview
(
path
),
"description"
:
self
.
find_description
(
path
),
"search_term"
:
self
.
search_terms_from_path
(
lora_on_disk
.
filename
),
"prompt"
:
json
.
dumps
(
f
"<lora:{
name
}:"
)
+
" + opts.extra_networks_default_multiplier + "
+
json
.
dumps
(
">"
),
"prompt"
:
json
.
dumps
(
f
"<lora:{
lora_on_disk.alias
}:"
)
+
" + opts.extra_networks_default_multiplier + "
+
json
.
dumps
(
">"
),
"local_preview"
:
f
"{path}.{shared.opts.samples_format}"
,
"metadata"
:
json
.
dumps
(
lora_on_disk
.
metadata
,
indent
=
4
)
if
lora_on_disk
.
metadata
else
None
,
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment