Upload ./README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -80,6 +80,21 @@ python sample_t2i.py --prompt "渔舟唱晚" --infer-mode trt
|
|
80 |
python sample_t2i.py --prompt "渔舟唱晚" --infer-mode trt --no-enhance
|
81 |
```
|
82 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
83 |
## ❓ Q&A
|
84 |
|
85 |
Please refer to the [Q&A](./QA.md) for more questions and answers about building the TensorRT Engine.
|
|
|
80 |
python sample_t2i.py --prompt "渔舟唱晚" --infer-mode trt --no-enhance
|
81 |
```
|
82 |
|
83 |
+
### 5. Notice
|
84 |
+
|
85 |
+
The TensorRT engine is designed to support following shapes of input for performance reasons.
|
86 |
+
In the future, we will verify and try to support arbitrary shapes.
|
87 |
+
|
88 |
+
```python
|
89 |
+
STANDARD_SHAPE = [
|
90 |
+
[(768, 768), (1024, 1024), (1280, 1280)], # 1:1
|
91 |
+
[(1024, 768), (1152, 864), (1280, 960)], # 4:3
|
92 |
+
[(768, 1024), (864, 1152), (960, 1280)], # 3:4
|
93 |
+
[(1280, 768)], # 16:9
|
94 |
+
[(768, 1280)], # 9:16
|
95 |
+
]
|
96 |
+
```
|
97 |
+
|
98 |
## ❓ Q&A
|
99 |
|
100 |
Please refer to the [Q&A](./QA.md) for more questions and answers about building the TensorRT Engine.
|