Это работает и с headless
и небездисплейный, и запустит окно с указанного размера вместо того, чтобы установить его после:
from selenium.webdriver import Firefox, FirefoxOptions
opts = FirefoxOptions()
opts.add_argument("--width=2560")
opts.add_argument("--height=1440")
driver = Firefox(options=opts)