from lyraSD import LyraSD | |
t2imodel = LyraSD("text2img", "./sd1.5-engine") | |
t2imodel.inference(prompt="A fantasy landscape, trending on artstation", use_super=True) | |
from PIL import Image | |
i2imodel = LyraSD("img2img", "./sd1.5-engine") | |
demo_img = Image.open("output/img2img_input.jpg") | |
i2imodel.inference(prompt="A fantasy landscape, trending on artstation", | |
image=demo_img, use_super=True) | |