Open liuzhiyong01 opened 1 week ago
Is this the complete repro code? When I run the above it shows me
TypeError: forward() missing 1 required positional argument: 'L_self_modules_pooler_modules_dense_parameters_bias_'
seems like it is missing some input.
Bug Description
when i use torch_tensorrt.compile transformer module with dynamic_shapes, it will occur this error,
To Reproduce
def test_compile_v1(): model = AutoModel.from_pretrained('bert-base-case', use_cache=False)
Enabled precision for TensorRT optimization
Environment
errors: