mirror of
https://github.com/vladmandic/automatic
synced 2026-09-17 08:19:11 +02:00
add /sdapi/v1/modules endpoint
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -99,6 +99,7 @@ class Api:
|
||||
self.add_api_route("/sdapi/v1/refresh-vae", endpoints.post_refresh_vae, methods=["POST"])
|
||||
self.add_api_route("/sdapi/v1/latents", endpoints.get_latent_history, methods=["GET"], response_model=List[str])
|
||||
self.add_api_route("/sdapi/v1/latents", endpoints.post_latent_history, methods=["POST"], response_model=int)
|
||||
self.add_api_route("/sdapi/v1/modules", endpoints.get_modules, methods=["GET"])
|
||||
|
||||
# lora api
|
||||
from modules.api import loras
|
||||
|
||||
@@ -156,6 +156,30 @@ def post_refresh_vae():
|
||||
shared.refresh_vaes()
|
||||
return {}
|
||||
|
||||
def get_modules():
|
||||
from modules import modelstats
|
||||
model = modelstats.analyze()
|
||||
if model is None:
|
||||
return {}
|
||||
model_obj = {
|
||||
'model': model.name,
|
||||
'type': model.type,
|
||||
'class': model.cls,
|
||||
'size': model.size,
|
||||
'mtime': str(model.mtime),
|
||||
'modules': []
|
||||
}
|
||||
for m in model.modules:
|
||||
model_obj['modules'].append({
|
||||
'class': m.cls,
|
||||
'params': m.params,
|
||||
'modules': m.modules,
|
||||
'quant': m.quant,
|
||||
'device': str(m.device),
|
||||
'dtype': str(m.dtype)
|
||||
})
|
||||
return model_obj
|
||||
|
||||
def get_extensions_list():
|
||||
from modules import extensions
|
||||
extensions.list_extensions()
|
||||
|
||||
Reference in New Issue
Block a user