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
44c37f94
Commit
44c37f94
authored
May 18, 2023
by
AUTOMATIC
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add messages about Loras that failed to load to UI
parent
cd8a510c
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
2 deletions
+8
-2
extensions-builtin/Lora/lora.py
extensions-builtin/Lora/lora.py
+7
-1
modules/processing.py
modules/processing.py
+1
-1
No files found.
extensions-builtin/Lora/lora.py
View file @
44c37f94
...
@@ -3,7 +3,7 @@ import re
...
@@ -3,7 +3,7 @@ import re
import
torch
import
torch
from
typing
import
Union
from
typing
import
Union
from
modules
import
shared
,
devices
,
sd_models
,
errors
,
scripts
from
modules
import
shared
,
devices
,
sd_models
,
errors
,
scripts
,
sd_hijack
metadata_tags_order
=
{
"ss_sd_model_name"
:
1
,
"ss_resolution"
:
2
,
"ss_clip_skip"
:
3
,
"ss_num_train_images"
:
10
,
"ss_tag_frequency"
:
20
}
metadata_tags_order
=
{
"ss_sd_model_name"
:
1
,
"ss_resolution"
:
2
,
"ss_clip_skip"
:
3
,
"ss_num_train_images"
:
10
,
"ss_tag_frequency"
:
20
}
...
@@ -211,6 +211,8 @@ def load_loras(names, multipliers=None):
...
@@ -211,6 +211,8 @@ def load_loras(names, multipliers=None):
loras_on_disk
=
[
available_lora_aliases
.
get
(
name
,
None
)
for
name
in
names
]
loras_on_disk
=
[
available_lora_aliases
.
get
(
name
,
None
)
for
name
in
names
]
failed_to_load_loras
=
[]
for
i
,
name
in
enumerate
(
names
):
for
i
,
name
in
enumerate
(
names
):
lora
=
already_loaded
.
get
(
name
,
None
)
lora
=
already_loaded
.
get
(
name
,
None
)
...
@@ -224,12 +226,16 @@ def load_loras(names, multipliers=None):
...
@@ -224,12 +226,16 @@ def load_loras(names, multipliers=None):
continue
continue
if
lora
is
None
:
if
lora
is
None
:
failed_to_load_loras
.
append
(
name
)
print
(
f
"Couldn't find Lora with name {name}"
)
print
(
f
"Couldn't find Lora with name {name}"
)
continue
continue
lora
.
multiplier
=
multipliers
[
i
]
if
multipliers
else
1.0
lora
.
multiplier
=
multipliers
[
i
]
if
multipliers
else
1.0
loaded_loras
.
append
(
lora
)
loaded_loras
.
append
(
lora
)
if
len
(
failed_to_load_loras
)
>
0
:
sd_hijack
.
model_hijack
.
comments
.
append
(
"Failed to find Loras: "
+
", "
.
join
(
failed_to_load_loras
))
def
lora_calc_updown
(
lora
,
module
,
target
):
def
lora_calc_updown
(
lora
,
module
,
target
):
with
torch
.
no_grad
():
with
torch
.
no_grad
():
...
...
modules/processing.py
View file @
44c37f94
...
@@ -808,7 +808,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
...
@@ -808,7 +808,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
images_list
=
output_images
,
images_list
=
output_images
,
seed
=
p
.
all_seeds
[
0
],
seed
=
p
.
all_seeds
[
0
],
info
=
infotext
(),
info
=
infotext
(),
comments
=
""
.
join
(
f
"
\n\n
{comment}
"
for
comment
in
comments
),
comments
=
""
.
join
(
f
"
{comment}
\n
"
for
comment
in
comments
),
subseed
=
p
.
all_subseeds
[
0
],
subseed
=
p
.
all_subseeds
[
0
],
index_of_first_image
=
index_of_first_image
,
index_of_first_image
=
index_of_first_image
,
infotexts
=
infotexts
,
infotexts
=
infotexts
,
...
...
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