32 lines
922 B
Python
32 lines
922 B
Python
import aiohttp
|
|
from bot.db import db, DBTables, decrypt
|
|
from rich import print
|
|
|
|
|
|
async def get_models():
|
|
endpoint = decrypt(db[DBTables.config].get('endpoint'))
|
|
try:
|
|
async with aiohttp.ClientSession() as session:
|
|
r = await session.get(endpoint + "/sdapi/v1/sd-models")
|
|
if r.status != 200:
|
|
return None
|
|
return [x["title"] for x in await r.json()]
|
|
except Exception as e:
|
|
print(e)
|
|
return None
|
|
|
|
|
|
async def set_model(model_name: str):
|
|
endpoint = decrypt(db[DBTables.config].get('endpoint'))
|
|
try:
|
|
async with aiohttp.ClientSession() as session:
|
|
r = await session.post(endpoint + "/sdapi/v1/options", json={
|
|
"sd_model_checkpoint": model_name
|
|
})
|
|
if r.status != 200:
|
|
return False
|
|
return True
|
|
except Exception as e:
|
|
print(e)
|
|
return False
|