English
Jarvis73 commited on
Commit
94871c6
·
verified ·
1 Parent(s): 0988d89

Upload ./README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +15 -0
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.