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
c48bbccf
Commit
c48bbccf
authored
Mar 04, 2023
by
Yea chen
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add: /sdapi/v1/scripts in API
API for get scripts list
parent
0cc0ee1b
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
17 additions
and
0 deletions
+17
-0
modules/api/api.py
modules/api/api.py
+13
-0
modules/api/models.py
modules/api/models.py
+4
-0
No files found.
modules/api/api.py
View file @
c48bbccf
...
@@ -150,6 +150,7 @@ class Api:
...
@@ -150,6 +150,7 @@ class Api:
self
.
add_api_route
(
"/sdapi/v1/train/embedding"
,
self
.
train_embedding
,
methods
=
[
"POST"
],
response_model
=
TrainResponse
)
self
.
add_api_route
(
"/sdapi/v1/train/embedding"
,
self
.
train_embedding
,
methods
=
[
"POST"
],
response_model
=
TrainResponse
)
self
.
add_api_route
(
"/sdapi/v1/train/hypernetwork"
,
self
.
train_hypernetwork
,
methods
=
[
"POST"
],
response_model
=
TrainResponse
)
self
.
add_api_route
(
"/sdapi/v1/train/hypernetwork"
,
self
.
train_hypernetwork
,
methods
=
[
"POST"
],
response_model
=
TrainResponse
)
self
.
add_api_route
(
"/sdapi/v1/memory"
,
self
.
get_memory
,
methods
=
[
"GET"
],
response_model
=
MemoryResponse
)
self
.
add_api_route
(
"/sdapi/v1/memory"
,
self
.
get_memory
,
methods
=
[
"GET"
],
response_model
=
MemoryResponse
)
self
.
add_api_route
(
"/sdapi/v1/scripts"
,
self
.
get_scripts_list
,
methods
=
[
"GET"
],
response_model
=
ScriptsList
)
def
add_api_route
(
self
,
path
:
str
,
endpoint
,
**
kwargs
):
def
add_api_route
(
self
,
path
:
str
,
endpoint
,
**
kwargs
):
if
shared
.
cmd_opts
.
api_auth
:
if
shared
.
cmd_opts
.
api_auth
:
...
@@ -174,6 +175,18 @@ class Api:
...
@@ -174,6 +175,18 @@ class Api:
script_idx
=
script_name_to_index
(
script_name
,
script_runner
.
selectable_scripts
)
script_idx
=
script_name_to_index
(
script_name
,
script_runner
.
selectable_scripts
)
script
=
script_runner
.
selectable_scripts
[
script_idx
]
script
=
script_runner
.
selectable_scripts
[
script_idx
]
return
script
,
script_idx
return
script
,
script_idx
def
get_scripts_list
(
self
):
t2ilist
=
[]
i2ilist
=
[]
for
a
in
scripts
.
scripts_txt2img
.
titles
:
t2ilist
.
append
(
str
(
a
.
lower
()))
for
b
in
scripts
.
scripts_img2img
.
titles
:
i2ilist
.
append
(
str
(
b
.
lower
()))
return
ScriptsList
(
txt2img
=
t2ilist
,
img2img
=
i2ilist
)
def
text2imgapi
(
self
,
txt2imgreq
:
StableDiffusionTxt2ImgProcessingAPI
):
def
text2imgapi
(
self
,
txt2imgreq
:
StableDiffusionTxt2ImgProcessingAPI
):
script
,
script_idx
=
self
.
get_script
(
txt2imgreq
.
script_name
,
scripts
.
scripts_txt2img
)
script
,
script_idx
=
self
.
get_script
(
txt2imgreq
.
script_name
,
scripts
.
scripts_txt2img
)
...
...
modules/api/models.py
View file @
c48bbccf
...
@@ -267,3 +267,7 @@ class EmbeddingsResponse(BaseModel):
...
@@ -267,3 +267,7 @@ class EmbeddingsResponse(BaseModel):
class
MemoryResponse
(
BaseModel
):
class
MemoryResponse
(
BaseModel
):
ram
:
dict
=
Field
(
title
=
"RAM"
,
description
=
"System memory stats"
)
ram
:
dict
=
Field
(
title
=
"RAM"
,
description
=
"System memory stats"
)
cuda
:
dict
=
Field
(
title
=
"CUDA"
,
description
=
"nVidia CUDA memory stats"
)
cuda
:
dict
=
Field
(
title
=
"CUDA"
,
description
=
"nVidia CUDA memory stats"
)
class
ScriptsList
(
BaseModel
):
txt2img
:
list
=
Field
(
default
=
None
,
title
=
"Txt2img"
,
description
=
"Titles of scripts (txt2img)"
)
img2img
:
list
=
Field
(
default
=
None
,
title
=
"Img2img"
,
description
=
"Titles of scripts (img2img)"
)
\ No newline at end of file
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