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
eb958095
Commit
eb958095
authored
May 09, 2023
by
AUTOMATIC
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
rework loras api
parent
7e02a00c
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
20 additions
and
38 deletions
+20
-38
extensions-builtin/Lora/lora.py
extensions-builtin/Lora/lora.py
+0
-6
extensions-builtin/Lora/scripts/api.py
extensions-builtin/Lora/scripts/api.py
+0
-31
extensions-builtin/Lora/scripts/lora_script.py
extensions-builtin/Lora/scripts/lora_script.py
+20
-1
No files found.
extensions-builtin/Lora/lora.py
View file @
eb958095
...
...
@@ -3,7 +3,6 @@ import os
import
re
import
torch
from
typing
import
Union
import
scripts.api
as
api
from
modules
import
shared
,
devices
,
sd_models
,
errors
,
scripts
...
...
@@ -449,8 +448,3 @@ available_lora_aliases = {}
loaded_loras
=
[]
list_available_loras
()
try
:
import
modules.script_callbacks
as
script_callbacks
script_callbacks
.
on_app_started
(
api
.
api
)
except
:
pass
\ No newline at end of file
extensions-builtin/Lora/scripts/api.py
deleted
100644 → 0
View file @
7e02a00c
from
fastapi
import
FastAPI
import
gradio
as
gr
import
json
import
os
import
lora
def
get_lora_prompts
(
path
):
directory
,
filename
=
os
.
path
.
split
(
path
)
name_without_ext
=
os
.
path
.
splitext
(
filename
)[
0
]
new_filename
=
name_without_ext
+
'.civitai.info'
try
:
new_path
=
os
.
path
.
join
(
directory
,
new_filename
)
if
os
.
path
.
exists
(
new_path
):
with
open
(
new_path
,
'r'
)
as
f
:
data
=
json
.
load
(
f
)
trained_words
=
data
.
get
(
'trainedWords'
,
[])
if
len
(
trained_words
)
>
0
:
result
=
','
.
join
(
trained_words
)
return
result
else
:
return
''
else
:
return
''
except
Exception
as
e
:
return
''
def
api
(
_
:
gr
.
Blocks
,
app
:
FastAPI
):
@
app
.
get
(
"/sdapi/v1/loras"
)
async
def
get_loras
():
return
[{
"name"
:
name
,
"path"
:
lora
.
available_loras
[
name
]
.
filename
,
"prompt"
:
get_lora_prompts
(
lora
.
available_loras
[
name
]
.
filename
)}
for
name
in
lora
.
available_loras
]
extensions-builtin/Lora/scripts/lora_script.py
View file @
eb958095
import
torch
import
gradio
as
gr
from
fastapi
import
FastAPI
import
lora
import
extra_networks_lora
import
ui_extra_networks_lora
from
modules
import
script_callbacks
,
ui_extra_networks
,
extra_networks
,
shared
def
unload
():
torch
.
nn
.
Linear
.
forward
=
torch
.
nn
.
Linear_forward_before_lora
torch
.
nn
.
Linear
.
_load_from_state_dict
=
torch
.
nn
.
Linear_load_state_dict_before_lora
...
...
@@ -60,3 +60,22 @@ shared.options_templates.update(shared.options_section(('extra_networks', "Extra
shared
.
options_templates
.
update
(
shared
.
options_section
((
'compatibility'
,
"Compatibility"
),
{
"lora_functional"
:
shared
.
OptionInfo
(
False
,
"Lora: use old method that takes longer when you have multiple Loras active and produces same results as kohya-ss/sd-webui-additional-networks extension"
),
}))
def
create_lora_json
(
obj
:
lora
.
LoraOnDisk
):
return
{
"name"
:
obj
.
name
,
"alias"
:
obj
.
alias
,
"path"
:
obj
.
filename
,
"metadata"
:
obj
.
metadata
,
}
def
api_loras
(
_
:
gr
.
Blocks
,
app
:
FastAPI
):
@
app
.
get
(
"/sdapi/v1/loras"
)
async
def
get_loras
():
return
[
create_lora_json
(
obj
)
for
obj
in
lora
.
available_loras
.
values
()]
script_callbacks
.
on_app_started
(
api_loras
)
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