Hugging Face's logo

TMElyralab
/
lyraSD

English
art
Stable Diffusion
Model card Files Community
3
moyanwang commited on
Commit
4d62d79
•
1 Parent(s): 75c208c

update demo

Browse files
Files changed (36) hide show
  1. .gitattributes +0 -0
  2. .gitignore +0 -0
  3. README.md +0 -0
  4. demo.py +1 -1
  5. lyraSD/__init__.py +0 -0
  6. lyraSD/__pycache__/__init__.cpython-38.pyc +0 -0
  7. lyraSD/__pycache__/inference.cpython-38.pyc +0 -0
  8. lyraSD/inference.py +1 -6
  9. lyraSD/muse_trt/__init__.py +0 -0
  10. lyraSD/muse_trt/__pycache__/__init__.cpython-38.pyc +0 -0
  11. lyraSD/muse_trt/__pycache__/models.cpython-38.pyc +0 -0
  12. lyraSD/muse_trt/__pycache__/sd_img2img.cpython-38.pyc +0 -0
  13. lyraSD/muse_trt/__pycache__/sd_text2img.cpython-38.pyc +0 -0
  14. lyraSD/muse_trt/__pycache__/super.cpython-38.pyc +0 -0
  15. lyraSD/muse_trt/__pycache__/utilities.cpython-38.pyc +0 -0
  16. lyraSD/muse_trt/libnvinfer_plugin.so +0 -0
  17. lyraSD/muse_trt/models.py +0 -0
  18. lyraSD/muse_trt/sd_img2img.py +0 -0
  19. lyraSD/muse_trt/sd_text2img.py +0 -0
  20. lyraSD/muse_trt/super.py +0 -0
  21. lyraSD/muse_trt/utilities.py +0 -0
  22. output/sd-img2img-0.jpg +0 -0
  23. output/sd-text2img-0.jpg +0 -0
  24. output/text2img_demo.jpg +0 -0
  25. sd1.5-engine/clip.plan +0 -0
  26. sd1.5-engine/feature_extractor/preprocessor_config.json +0 -0
  27. sd1.5-engine/scheduler/scheduler_config.json +0 -0
  28. sd1.5-engine/superx4.plan +0 -0
  29. sd1.5-engine/text_encoder/config.json +0 -0
  30. sd1.5-engine/tokenizer/merges.txt +0 -0
  31. sd1.5-engine/tokenizer/special_tokens_map.json +0 -0
  32. sd1.5-engine/tokenizer/tokenizer_config.json +0 -0
  33. sd1.5-engine/tokenizer/vocab.json +0 -0
  34. sd1.5-engine/unet_fp16.plan +0 -0
  35. sd1.5-engine/vae-decoder.plan +0 -0
  36. sd1.5-engine/vae/config.json +0 -0
.gitattributes CHANGED
File without changes
.gitignore CHANGED
File without changes
README.md CHANGED
File without changes
demo.py CHANGED
@@ -1,7 +1,7 @@
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
 
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=False)
5
 
6
 
7
  from PIL import Image
lyraSD/__init__.py CHANGED
File without changes
lyraSD/__pycache__/__init__.cpython-38.pyc CHANGED
Binary files a/lyraSD/__pycache__/__init__.cpython-38.pyc and b/lyraSD/__pycache__/__init__.cpython-38.pyc differ
 
lyraSD/__pycache__/inference.cpython-38.pyc CHANGED
Binary files a/lyraSD/__pycache__/inference.cpython-38.pyc and b/lyraSD/__pycache__/inference.cpython-38.pyc differ
 
lyraSD/inference.py CHANGED
@@ -35,7 +35,7 @@ class LyraSD(object):
35
  save_dir="./output",
36
  save_basename="sd-",
37
  negative_prompts='',
38
- strength=0.3,
39
  height=None,
40
  width =None,
41
  num_images_per_prompt=1,
@@ -56,11 +56,6 @@ class LyraSD(object):
56
  if width is None:
57
  width = self.o_width
58
 
59
- # this version model doen't support batch mode.
60
- if not isinstance(prompt, list):
61
- prompt = [prompt]
62
- if len(prompt) > 1:
63
- raise ValueError("current model dosen't support multi prompts")
64
  if self.sd_mode=="text2img":
65
  result_image = self.pipeline(prompt=prompt, negative_prompt=negative_prompts,
66
  num_inference_steps= num_inference_steps,
 
35
  save_dir="./output",
36
  save_basename="sd-",
37
  negative_prompts='',
38
+ strength=0.8,
39
  height=None,
40
  width =None,
41
  num_images_per_prompt=1,
 
56
  if width is None:
57
  width = self.o_width
58
 
 
 
 
 
 
59
  if self.sd_mode=="text2img":
60
  result_image = self.pipeline(prompt=prompt, negative_prompt=negative_prompts,
61
  num_inference_steps= num_inference_steps,
lyraSD/muse_trt/__init__.py CHANGED
File without changes
lyraSD/muse_trt/__pycache__/__init__.cpython-38.pyc CHANGED
Binary files a/lyraSD/muse_trt/__pycache__/__init__.cpython-38.pyc and b/lyraSD/muse_trt/__pycache__/__init__.cpython-38.pyc differ
 
lyraSD/muse_trt/__pycache__/models.cpython-38.pyc CHANGED
Binary files a/lyraSD/muse_trt/__pycache__/models.cpython-38.pyc and b/lyraSD/muse_trt/__pycache__/models.cpython-38.pyc differ
 
lyraSD/muse_trt/__pycache__/sd_img2img.cpython-38.pyc CHANGED
Binary files a/lyraSD/muse_trt/__pycache__/sd_img2img.cpython-38.pyc and b/lyraSD/muse_trt/__pycache__/sd_img2img.cpython-38.pyc differ
 
lyraSD/muse_trt/__pycache__/sd_text2img.cpython-38.pyc CHANGED
Binary files a/lyraSD/muse_trt/__pycache__/sd_text2img.cpython-38.pyc and b/lyraSD/muse_trt/__pycache__/sd_text2img.cpython-38.pyc differ
 
lyraSD/muse_trt/__pycache__/super.cpython-38.pyc CHANGED
Binary files a/lyraSD/muse_trt/__pycache__/super.cpython-38.pyc and b/lyraSD/muse_trt/__pycache__/super.cpython-38.pyc differ
 
lyraSD/muse_trt/__pycache__/utilities.cpython-38.pyc CHANGED
Binary files a/lyraSD/muse_trt/__pycache__/utilities.cpython-38.pyc and b/lyraSD/muse_trt/__pycache__/utilities.cpython-38.pyc differ
 
lyraSD/muse_trt/libnvinfer_plugin.so CHANGED
File without changes
lyraSD/muse_trt/models.py CHANGED
File without changes
lyraSD/muse_trt/sd_img2img.py CHANGED
File without changes
lyraSD/muse_trt/sd_text2img.py CHANGED
File without changes
lyraSD/muse_trt/super.py CHANGED
File without changes
lyraSD/muse_trt/utilities.py CHANGED
File without changes
output/sd-img2img-0.jpg CHANGED
output/sd-text2img-0.jpg CHANGED
output/text2img_demo.jpg CHANGED
sd1.5-engine/clip.plan CHANGED
File without changes
sd1.5-engine/feature_extractor/preprocessor_config.json CHANGED
File without changes
sd1.5-engine/scheduler/scheduler_config.json CHANGED
File without changes
sd1.5-engine/superx4.plan CHANGED
File without changes
sd1.5-engine/text_encoder/config.json CHANGED
File without changes
sd1.5-engine/tokenizer/merges.txt CHANGED
File without changes
sd1.5-engine/tokenizer/special_tokens_map.json CHANGED
File without changes
sd1.5-engine/tokenizer/tokenizer_config.json CHANGED
File without changes
sd1.5-engine/tokenizer/vocab.json CHANGED
File without changes
sd1.5-engine/unet_fp16.plan CHANGED
File without changes
sd1.5-engine/vae-decoder.plan CHANGED
File without changes
sd1.5-engine/vae/config.json CHANGED
File without changes