Skip to content

Commit 08a6ef9

Browse files
committed
Add kNATIVE_INSTANCENORM flag to demoDiffusion
Signed-off-by: Kevin Chen <kevinch@nvidia.com>
1 parent 0d2bf98 commit 08a6ef9

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

demo/Diffusion/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ diffusers 0.14.0
4848
onnx 1.13.1
4949
onnx-graphsurgeon 0.3.26
5050
onnxruntime 1.14.1
51-
polygraphy 0.44.2
52-
tensorrt 8.6.0.10
51+
polygraphy 0.47.1
52+
tensorrt 8.6.1.6
5353
tokenizers 0.13.2
5454
torch 1.13.0
5555
transformers 4.26.1

demo/Diffusion/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ onnx==1.13.1
99
onnxruntime==1.14.1
1010
--extra-index-url https://pypi.ngc.nvidia.com
1111
onnx-graphsurgeon==0.3.26
12-
polygraphy==0.44.2
12+
polygraphy==0.47.1
1313
scipy
1414
torch<2.0.0
1515
transformers==4.26.1

demo/Diffusion/utilities.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ def build(self, onnx_path, fp16, input_profile=None, enable_refit=False, enable_
210210
config_kwargs['tactic_sources'] = []
211211

212212
engine = engine_from_network(
213-
network_from_onnx_path(onnx_path),
213+
network_from_onnx_path(onnx_path, flags=[trt.OnnxParserFlag.NATIVE_INSTANCENORM]),
214214
config=CreateConfig(fp16=fp16,
215215
refittable=enable_refit,
216216
profiles=[p],

0 commit comments

Comments
 (0)