miaoshouai-assistant/scripts/main.py

27 lines
582 B
Python

import modules
import modules.scripts as scripts
from scripts.assistant.miaoshou import MiaoShouAssistant
assistant = MiaoShouAssistant()
class MiaoshouScript(scripts.Script):
def __init__(self) -> None:
super().__init__()
def title(self):
return "Miaoshou Assistant"
def show(self, is_img2img):
return scripts.AlwaysVisible
def ui(self, is_img2img):
return ()
def postprocess(self, p, processed):
assistant.release_mem()
return None
modules.script_callbacks.on_ui_tabs(assistant.on_event_ui_tabs_opened)