В последнее время эта проблема возникла для
Причина в том, что размер изображения слишком велик.
Поэтому уменьшите размер окна, например, до 768x1024
chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument('--headless')
...
chrome_options.add_argument('--window-size=768x1024')
И если Вы делаете снимки экрана полной страницы - ограничьте высоту
width = driver.execute_script('return document.body.parentNode.scrollWidth')
height = driver.execute_script('return document.body.parentNode.scrollHeight')
height = height if height < 10000 else 10000
driver.set_window_size(width, height)
driver.save_screenshot(image_path)