/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py:895: UserWarning: Unable to import torchvision related libraries.: No module named 'torchvision'. Please install torchvision lib in order to lower stochastic_depth warnings.warn( Unable to import quantization op. Please install modelopt library (https://github.com/NVIDIA/TensorRT-Model-Optimizer?tab=readme-ov-file#installation) to add support for compiling quantized models [12/19/2024-19:02:20] [TRT] [W] Functionality provided through tensorrt.plugin module is experimental. Loading pipeline components...: 0%| | 0/7 [00:00 trt_model = torch_tensorrt.dynamo.compile( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch_tensorrt/dynamo/_compiler.py", line 666, in compile exported_program = exported_program.run_decompositions( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/export/exported_program.py", line 128, in wrapper return fn(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/export/exported_program.py", line 1268, in run_decompositions return _decompose_exported_program( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/export/exported_program.py", line 742, in _decompose_exported_program gm, new_graph_signature = _decompose_and_get_gm_with_new_signature_constants( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/export/exported_program.py", line 454, in _decompose_and_get_gm_with_new_signature_constants aten_export_artifact = _export_to_aten_ir( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/export/_trace.py", line 741, in _export_to_aten_ir gm, graph_signature = transform(aot_export_module)( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/aot_autograd.py", line 1332, in aot_export_module fx_g, metadata, in_spec, out_spec = _aot_export_function( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/aot_autograd.py", line 1571, in _aot_export_function fx_g, meta = create_aot_dispatcher_function( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/aot_autograd.py", line 580, in create_aot_dispatcher_function return _create_aot_dispatcher_function( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/aot_autograd.py", line 830, in _create_aot_dispatcher_function compiled_fn, fw_metadata = compiler_fn( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/jit_compile_runtime_wrappers.py", line 118, in aot_dispatch_export graph, _, _ = aot_dispatch_base_graph( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/dispatch_and_compile_graph.py", line 153, in aot_dispatch_base_graph fw_module = _create_graph( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/dispatch_and_compile_graph.py", line 55, in _create_graph fx_g = make_fx( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 2196, in wrapped return make_fx_tracer.trace(f, *args) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 2134, in trace return self._trace_inner(f, *args) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 2105, in _trace_inner t = dispatch_trace( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_compile.py", line 32, in inner return disable_fn(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 751, in _fn return fn(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 1138, in dispatch_trace graph = tracer.trace(root, concrete_args) # type: ignore[arg-type] File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 1694, in trace res = super().trace(root, concrete_args) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/_symbolic_trace.py", line 843, in trace (self.create_arg(fn(*args)),), File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 1193, in wrapped out = f(*tensors) # type:ignore[call-arg] File "", line 1, in File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/traced_function_transforms.py", line 693, in inner_fn outs = fn(*args) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/traced_function_transforms.py", line 413, in _functionalized_f_helper f_outs = fn(*f_args) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/traced_function_transforms.py", line 78, in inner_fn outs = fn(*args) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/utils.py", line 184, in flat_fn tree_out = fn(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/traced_function_transforms.py", line 875, in functional_call out = PropagateUnbackedSymInts(mod).run( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/interpreter.py", line 167, in run self.env[node] = self.run_node(node) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py", line 6795, in run_node result = super().run_node(n) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/interpreter.py", line 230, in run_node return getattr(self, n.op)(n.target, args, kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/interpreter.py", line 310, in call_function return target(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_ops.py", line 722, in __call__ return self._op(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 1241, in __torch_function__ return func(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_ops.py", line 722, in __call__ return self._op(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_subclasses/functional_tensor.py", line 527, in __torch_dispatch__ outs_unwrapped = func._op_dk( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/utils/_stats.py", line 20, in wrapper return fn(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 1343, in __torch_dispatch__ return proxy_call(self, func, self.pre_dispatch, args, kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/fx/experimental/proxy_tensor.py", line 912, in proxy_call out = func(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_ops.py", line 722, in __call__ return self._op(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/utils/_stats.py", line 20, in wrapper return fn(*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_subclasses/fake_tensor.py", line 1276, in __torch_dispatch__ return self.dispatch(func, types, args, kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_subclasses/fake_tensor.py", line 1817, in dispatch return self._cached_dispatch_impl(func, types, args, kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_subclasses/fake_tensor.py", line 1387, in _cached_dispatch_impl output = self._dispatch_impl(func, types, args, kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_subclasses/fake_tensor.py", line 2291, in _dispatch_impl decomposition_table[func](*args, **kwargs) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_refs/__init__.py", line 4674, in view return _reshape_view_helper(a, *shape, allow_copy=False) File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/torch/_refs/__init__.py", line 3807, in _reshape_view_helper raise ValueError(msg) ValueError: Cannot view a tensor with shape torch.Size([s6, s2 + 4096, 24, 128]) and strides (3072*s2 + 12582912, 128, 128*s2 + 524288, 1) as a tensor with shape (s1, (s6*(s2 + 4096)//s1), 3072)! While executing %view_52 : [num_users=1] = call_function[target=torch.ops.aten.view.default](args = (%transpose_10, [%sym_size_int_63, -1, 3072]), kwargs = {}) Original traceback: File "/work/TensorRT/examples/dynamo/run_2.py", line 48, in forward return self.module.forward( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/diffusers/models/transformers/transformer_flux.py", line 438, in forward hidden_states = block( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/diffusers/models/transformers/transformer_flux.py", line 119, in forward attn_output = self.attn( File "/root/.pyenv/versions/3.10.16/lib/python3.10/site-packages/diffusers/models/attention_processor.py", line 490, in forward return self.processor(