From 2c8cf941504ef7852e68516b343573a33fb3980b Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Sat, 25 Jan 2025 00:37:41 +0800
Subject: [PATCH] Update accel_sdxl_gen_img.py

---
 accel_sdxl_gen_img.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py
index 1c7c871f..e926e466 100644
--- a/accel_sdxl_gen_img.py
+++ b/accel_sdxl_gen_img.py
@@ -14,6 +14,7 @@ import os
 import random
 import re
 import gc
+import sdxl_train_network
 
 import diffusers
 import numpy as np
@@ -2860,8 +2861,8 @@ def main(args):
 
 
 def setup_parser() -> argparse.ArgumentParser:
-    parser = train_network.setup_parser()
-    sdxl_train_util.add_sdxl_training_arguments(parser)
+    parser = sdxl_train_network.setup_parser()
+    #sdxl_train_util.add_sdxl_training_arguments(parser)
     add_logging_arguments(parser)
 
     parser.add_argument("--prompt", type=str, default=None, help="prompt / プロンプト")