Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in / Register
Toggle navigation
S
stable-diffusion-webui
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Administrator
stable-diffusion-webui
Commits
500d9a32
Commit
500d9a32
authored
Jan 21, 2023
by
AUTOMATIC
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add --lora-dir commandline option
parent
4a8fe096
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
11 additions
and
6 deletions
+11
-6
lora.py
extensions-builtin/Lora/lora.py
+4
-5
preload.py
extensions-builtin/Lora/preload.py
+6
-0
ui_extra_networks_lora.py
extensions-builtin/Lora/ui_extra_networks_lora.py
+1
-1
No files found.
extensions-builtin/Lora/lora.py
View file @
500d9a32
...
@@ -177,12 +177,12 @@ def lora_Conv2d_forward(self, input):
...
@@ -177,12 +177,12 @@ def lora_Conv2d_forward(self, input):
def
list_available_loras
():
def
list_available_loras
():
available_loras
.
clear
()
available_loras
.
clear
()
os
.
makedirs
(
lora_dir
,
exist_ok
=
True
)
os
.
makedirs
(
shared
.
cmd_opts
.
lora_dir
,
exist_ok
=
True
)
candidates
=
\
candidates
=
\
glob
.
glob
(
os
.
path
.
join
(
lora_dir
,
'**/*.pt'
),
recursive
=
True
)
+
\
glob
.
glob
(
os
.
path
.
join
(
shared
.
cmd_opts
.
lora_dir
,
'**/*.pt'
),
recursive
=
True
)
+
\
glob
.
glob
(
os
.
path
.
join
(
lora_dir
,
'**/*.safetensors'
),
recursive
=
True
)
+
\
glob
.
glob
(
os
.
path
.
join
(
shared
.
cmd_opts
.
lora_dir
,
'**/*.safetensors'
),
recursive
=
True
)
+
\
glob
.
glob
(
os
.
path
.
join
(
lora_dir
,
'**/*.ckpt'
),
recursive
=
True
)
glob
.
glob
(
os
.
path
.
join
(
shared
.
cmd_opts
.
lora_dir
,
'**/*.ckpt'
),
recursive
=
True
)
for
filename
in
sorted
(
candidates
):
for
filename
in
sorted
(
candidates
):
if
os
.
path
.
isdir
(
filename
):
if
os
.
path
.
isdir
(
filename
):
...
@@ -193,7 +193,6 @@ def list_available_loras():
...
@@ -193,7 +193,6 @@ def list_available_loras():
available_loras
[
name
]
=
LoraOnDisk
(
name
,
filename
)
available_loras
[
name
]
=
LoraOnDisk
(
name
,
filename
)
lora_dir
=
os
.
path
.
join
(
shared
.
models_path
,
"Lora"
)
available_loras
=
{}
available_loras
=
{}
loaded_loras
=
[]
loaded_loras
=
[]
...
...
extensions-builtin/Lora/preload.py
0 → 100644
View file @
500d9a32
import
os
from
modules
import
paths
def
preload
(
parser
):
parser
.
add_argument
(
"--lora-dir"
,
type
=
str
,
help
=
"Path to directory with Lora networks."
,
default
=
os
.
path
.
join
(
paths
.
models_path
,
'Lora'
))
extensions-builtin/Lora/ui_extra_networks_lora.py
View file @
500d9a32
...
@@ -31,5 +31,5 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
...
@@ -31,5 +31,5 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
}
}
def
allowed_directories_for_previews
(
self
):
def
allowed_directories_for_previews
(
self
):
return
[
lora
.
lora_dir
]
return
[
shared
.
cmd_opts
.
lora_dir
]
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