Issue while generating output with GFPGAN tensorrt

[alakhsharmaa]

trt_results, trt_time = run_inference(trt_model, images)

---

RuntimeError Traceback (most recent call last)
in <cell line: 1>()
----> 1 trt_results, trt_time = run_inference_in_batches(trt_model, driving_images[:2])

2 frames
/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py in _call_impl(self, *args, **kwargs)
1745 or _global_backward_pre_hooks or _global_backward_hooks
1746 or _global_forward_hooks or _global_forward_pre_hooks):
-> 1747 return forward_call(*args, **kwargs)
1748
1749 result = None

RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript (most recent call last):
RuntimeError: [Error thrown at core/runtime/execute_engine.cpp:231] Expected compiled_engine->exec_ctx->setInputShape(name.c_str(), dims) to be true but got false
Error while setting the input shape