English
art
Stable Diffusion
moyanwang commited on
Commit
806e745
1 Parent(s): 25dabe4
Files changed (1) hide show
  1. demo.py +11 -0
demo.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from lyraSD import LyraSD
2
+
3
+ t2imodel = LyraSD("text2img", "./sd1.5-engine")
4
+ t2imodel.inference(prompt="a red ballon flying in the sky")
5
+
6
+
7
+ from PIL import Image
8
+ i2imodel = LyraSD("img2img", "./sd1.5-engine")
9
+ demo_img = Image.open("output/text2img_demo.jpg")
10
+ i2imodel.inference(prompt="comic style", image=demo_img)
11
+