Add dynamic False in transfoormers example

pytorch · Jan 20, 2024 · bf81c97 · bf81c97
1 parent d5868ba
commit bf81c97
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/dynamo/torch_compile_transformers_example.py b/examples/dynamo/torch_compile_transformers_example.py
@@ -63,6 +63,7 @@
     model,
     backend="torch_tensorrt",
     options=compilation_kwargs,
+    dynamic=False,
 )
 optimized_model(*inputs)