Commit 34a82a34 authored by Sayo's avatar Sayo

Add api method to get LoRA models

parent 6a5901a3
......@@ -2,7 +2,9 @@ import glob
import os
import re
import torch
from typing import Union
from typing import Union, List, Optional
from fastapi import FastAPI
import gradio as gr
from modules import shared, devices, sd_models, errors, scripts
......@@ -443,9 +445,19 @@ def infotext_pasted(infotext, params):
if added:
params["Prompt"] += "\n" + "".join(added)
def api(_: gr.Blocks, app: FastAPI):
@app.get("/sdapi/v1/loras")
async def getloras():
return [{"name": name, "path": available_loras[name].filename, "prompt": ""} for name in available_loras]
available_loras = {}
available_lora_aliases = {}
loaded_loras = []
list_available_loras()
try:
import modules.script_callbacks as script_callbacks
script_callbacks.on_app_started(api)
except:
pass
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment