English
art
Stable Diffusion
lyraSD / README.md
bigmoyan's picture
Update README.md
25dabe4
|
raw
history blame
1 kB
metadata
license: creativeml-openrail-m
language:
  - zh
  - en
library_name: diffusers
pipeline_tag: text-to-image
tags:
  - art
  - tensorRT

Model Card for fast_sd_1.5

fast_sd_1.5 is a tensorRT implementation of Stable diffusion model, with stable 1.5 weights. Main features

  • weights: original stable diffusion 1.4 weights
  • input image size range: 256~640 (min 256x256, max 640x640)
  • device requirements: Nvidia Ampere architecture or newer (A100)
  • super-resultion: 4x by default, optional.

fast_sd_1.5 is a part of TME lyralab Fast Model Plan.

Model Sources [optional]

Uses

from lyraSD import LyraSD

t2imodel = LyraSD("text2img", "./sd1.5-engine")
t2imodel.inference(prompt="a red ballon flying in the sky")


from PIL import Image
i2imodel = LyraSD("img2img", "./sd1.5-engine")
demo_img = Image.open("output/text2img_demo.jpg")
i2imodel.inference(prompt="comic style", image=demo_img)