StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add api method to get LoRA models

Browse Source
This commit is contained in:
Sayo 2023-05-08 19:55:05 +08:00
parent 6a5901a3fd
commit 34a82a345a
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
extensions-builtin/Lora/lora.py
View File

@ -2,7 +2,9 @@ import glob
import os import os
import re import re
import torch 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 from modules import shared, devices, sd_models, errors, scripts
@ -443,9 +445,19 @@ def infotext_pasted(infotext, params):
if added: if added:
params["Prompt"] += "\n" + "".join(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_loras = {}
available_lora_aliases = {} available_lora_aliases = {}
loaded_loras = [] loaded_loras = []
list_available_loras() list_available_loras()
try:
import modules.script_callbacks as script_callbacks
script_callbacks.on_app_started(api)
except:
pass