| 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) | |