layerdiffusion
commited on
Commit
·
1b371ad
1
Parent(s):
0e5fe5d
better gradio args for win
Browse files- demo_gradio.py +6 -1
demo_gradio.py
CHANGED
|
@@ -31,9 +31,13 @@ from diffusers_helper.bucket_tools import find_nearest_bucket
|
|
| 31 |
parser = argparse.ArgumentParser()
|
| 32 |
parser.add_argument('--share', action='store_true')
|
| 33 |
parser.add_argument("--server", type=str, default='0.0.0.0')
|
| 34 |
-
parser.add_argument("--port", type=int,
|
|
|
|
| 35 |
args = parser.parse_args()
|
| 36 |
|
|
|
|
|
|
|
|
|
|
| 37 |
print(args)
|
| 38 |
|
| 39 |
free_mem_gb = get_cuda_free_memory_gb(gpu)
|
|
@@ -396,4 +400,5 @@ block.launch(
|
|
| 396 |
server_name=args.server,
|
| 397 |
server_port=args.port,
|
| 398 |
share=args.share,
|
|
|
|
| 399 |
)
|
|
|
|
| 31 |
parser = argparse.ArgumentParser()
|
| 32 |
parser.add_argument('--share', action='store_true')
|
| 33 |
parser.add_argument("--server", type=str, default='0.0.0.0')
|
| 34 |
+
parser.add_argument("--port", type=int, required=False)
|
| 35 |
+
parser.add_argument("--inbrowser", action='store_true')
|
| 36 |
args = parser.parse_args()
|
| 37 |
|
| 38 |
+
# for win desktop probably use --server 127.0.0.1 --inbrowser
|
| 39 |
+
# For linux server probably use --server 127.0.0.1 or do not use any cmd flags
|
| 40 |
+
|
| 41 |
print(args)
|
| 42 |
|
| 43 |
free_mem_gb = get_cuda_free_memory_gb(gpu)
|
|
|
|
| 400 |
server_name=args.server,
|
| 401 |
server_port=args.port,
|
| 402 |
share=args.share,
|
| 403 |
+
inbrowser=args.inbrowser,
|
| 404 |
)
|