From 34ee4f22a56eb1b3055df2596b6c2b57434a405b Mon Sep 17 00:00:00 2001
From: zhaoyafan <zhaoyafan@foxmail.com>
Date: Sun, 7 May 2023 17:00:55 +0800
Subject: [PATCH] Commit.

---
 Lib/Driver.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Lib/Driver.py b/Lib/Driver.py
index 30d31e5..f530ce6 100644
--- a/Lib/Driver.py
+++ b/Lib/Driver.py
@@ -622,11 +622,12 @@ class Browser(browser_webdriver):
             options.add_argument(i)
         # Set headless mode.
         if self.is_linux or headless:
-            options.add_argument('--headless=new')
+            options.add_argument('--headless=new' if browser_choose != 1 else '--headless')
         # Set no-sandbox mode.
         if self.is_linux:
             options.add_argument('--no-sandbox')
             options.add_argument('--disable-dev-shm-usage')
+        if self.is_linux and browser_choose != 1:
             options.add_argument('--disable-gpu')
         # Set language of browser, default is zh-CN.
         if lang: