bigmoyan commited on
Commit
5e81e3a
1 Parent(s): d58d7c2

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +2 -2
demo.py CHANGED
@@ -1,11 +1,11 @@
1
  from lyraSD import LyraSD
2
 
3
- t2imodel = LyraSD("text2img", "./sd1.5-engine")
4
  t2imodel.inference(prompt="A fantasy landscape, trending on artstation", use_super=True)
5
 
6
 
7
  from PIL import Image
8
- i2imodel = LyraSD("img2img", "./sd1.5-engine")
9
  demo_img = Image.open("output/img2img_input.jpg")
10
  i2imodel.inference(prompt="A fantasy landscape, trending on artstation",
11
  image=demo_img, use_super=True)
 
1
  from lyraSD import LyraSD
2
 
3
+ t2imodel = LyraSD("text2img", "./sd1.4-engine")
4
  t2imodel.inference(prompt="A fantasy landscape, trending on artstation", use_super=True)
5
 
6
 
7
  from PIL import Image
8
+ i2imodel = LyraSD("img2img", "./sd1.4-engine")
9
  demo_img = Image.open("output/img2img_input.jpg")
10
  i2imodel.inference(prompt="A fantasy landscape, trending on artstation",
11
  image=demo_img, use_super=True)