File size: 985 Bytes
ae89015
 
9ced76e
 
 
 
 
 
 
 
ae89015
76d6b62
9ced76e
76d6b62
9ced76e
 
76d6b62
 
 
 
 
9ced76e
 
 
 
 
 
 
 
76d6b62
 
 
9ced76e
 
76d6b62
9ced76e
76d6b62
 
 
9ced76e
76d6b62
 
 
 
9ced76e
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
---
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]

- **Repository:** [https://huggingface.co/runwayml/stable-diffusion-v1-5]

## Uses

```python
from fast_sd_1.5 import FSD
from PIL import Image


inference_model = FSD(plan="path/to/plan/dir", superres="4x")

# text input
text = "A red ballon flying in the sky"
output_img = inference_model.infer(text=text)

# img input
img = Image.open("test.jpg")
output_img = inference_model.infer(img=img)
```