lyraSD / demo.py
bigmoyan's picture
Update demo.py
5e81e3a
raw
history blame
No virus
411 Bytes
from lyraSD import LyraSD
t2imodel = LyraSD("text2img", "./sd1.4-engine")
t2imodel.inference(prompt="A fantasy landscape, trending on artstation", use_super=True)
from PIL import Image
i2imodel = LyraSD("img2img", "./sd1.4-engine")
demo_img = Image.open("output/img2img_input.jpg")
i2imodel.inference(prompt="A fantasy landscape, trending on artstation",
image=demo_img, use_super=True)