from lyraSD import LyraSD

t2imodel = LyraSD("text2img", "./sd1.5-engine")
t2imodel.inference(prompt="a red ballon flying in the sky")


from PIL import Image
i2imodel = LyraSD("img2img", "./sd1.5-engine")
demo_img = Image.open("output/text2img_demo.jpg")
i2imodel.inference(prompt="comic style", image=demo_img)