mirror of
https://github.com/vladmandic/sdnext.git
synced 2026-01-27 15:02:48 +03:00
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from typing import List
|
|
from fastapi.exceptions import HTTPException
|
|
|
|
|
|
def get_lora(lora: str) -> dict:
|
|
from modules.lora import lora_load
|
|
if lora not in lora_load.available_networks:
|
|
raise HTTPException(status_code=404, detail=f"Lora '{lora}' not found")
|
|
obj = lora_load.available_networks[lora]
|
|
return obj.__dict__
|
|
|
|
|
|
def get_loras():
|
|
from modules.lora import network, lora_load
|
|
def create_lora_json(obj: network.NetworkOnDisk):
|
|
return { "name": obj.name, "alias": obj.alias, "path": obj.filename, "metadata": obj.metadata }
|
|
return [create_lora_json(obj) for obj in lora_load.available_networks.values()]
|
|
|
|
|
|
def post_refresh_loras():
|
|
from modules.lora import lora_load
|
|
return lora_load.list_available_networks()
|
|
|
|
|
|
def register_api():
|
|
from modules.shared import api
|
|
api.add_api_route("/sdapi/v1/lora", get_lora, methods=["GET"], response_model=dict)
|
|
api.add_api_route("/sdapi/v1/loras", get_loras, methods=["GET"], response_model=List[dict])
|
|
api.add_api_route("/sdapi/v1/refresh-loras", post_refresh_loras, methods=["POST"])
|