going higher than 768x768 gives an error but if batch size is 2 then one image gens correctly

[Dargunsh]

[E] 3: [executionContext.cpp::nvinfer1::rt::ExecutionContext::validateInputBindings::2046] Error Code 3: API Usage Error (Parameter check failed at: runtime/api/executionContext.cpp::nvinfer1::rt::ExecutionContext::validateInputBindings::2046, condition: profileMinDims.d[i] <= dimensions.d[i]. Supplied binding dimension [2,77,768] for bindings[2] exceed min ~ max range at index 1, maximum dimension in profile is 231, minimum dimension in profile is 154, but supplied dimension is 77.