tharms commited on
Commit
2d7ebb1
·
1 Parent(s): 079b935
args_manager.py CHANGED
@@ -37,7 +37,6 @@ args_parser.parser.add_argument("--always-download-new-model", action='store_tru
37
  args_parser.parser.set_defaults(
38
  disable_cuda_malloc=True,
39
  in_browser=True,
40
- api_mode=False,
41
  port=None
42
  )
43
 
 
37
  args_parser.parser.set_defaults(
38
  disable_cuda_malloc=True,
39
  in_browser=True,
 
40
  port=None
41
  )
42
 
language/en.json CHANGED
@@ -1,33 +1,14 @@
1
  {
2
- "# <center>zStudios Generative Media AI running on CPU</center>": "# <center>zStudios Generative Media AI running on CPU</center>",
3
- "DALL-E": "DALL-E",
4
- "Model": "Model",
5
- "dall-e-3": "dall-e-3",
6
- "dall-e-2": "dall-e-2",
7
- "Quality": "Quality",
8
- "standard": "standard",
9
- "hd": "hd",
10
- "Size": "Size",
11
- "1024x1024": "1024x1024",
12
- "256x256": "256x256",
13
- "512x512": "512x512",
14
- "1792x1024": "1792x1024",
15
- "1024x1792": "1024x1792",
16
- "Prompt": "Prompt",
17
- "Run": "Run",
18
- "Image Output": "Image Output",
19
- "Fooocus": "Fooocus",
20
- "Deactivated": "Deactivated",
21
  "Preview": "Preview",
22
- "Finished Images": "Finished Images",
23
  "Gallery": "Gallery",
24
  "Generate": "Generate",
25
- "Load Parameters": "Load Parameters",
26
  "Skip": "Skip",
27
  "Stop": "Stop",
28
  "Input Image": "Input Image",
29
  "Advanced": "Advanced",
30
  "Upscale or Variation": "Upscale or Variation",
 
 
31
  "Drag above image to here": "Drag above image to here",
32
  "Upscale or Variation:": "Upscale or Variation:",
33
  "Disabled": "Disabled",
@@ -36,105 +17,50 @@
36
  "Upscale (1.5x)": "Upscale (1.5x)",
37
  "Upscale (2x)": "Upscale (2x)",
38
  "Upscale (Fast 2x)": "Upscale (Fast 2x)",
39
- "<a href=\"https://github.com/lllyasviel/Fooocus/discussions/390\" target=\"_blank\">\ud83d\udcd4 Document</a>": "<a href=\"https://github.com/lllyasviel/Fooocus/discussions/390\" target=\"_blank\">\ud83d\udcd4 Document</a>",
40
- "Image Prompt": "Image Prompt",
41
  "Image": "Image",
42
  "Stop At": "Stop At",
43
  "Weight": "Weight",
44
  "Type": "Type",
45
- "ImagePrompt": "ImagePrompt",
46
  "PyraCanny": "PyraCanny",
47
  "CPDS": "CPDS",
48
- "FaceSwap": "FaceSwap",
49
- "* \"Image Prompt\" is powered by Fooocus Image Mixture Engine (v1.0.1). <a href=\"https://github.com/lllyasviel/Fooocus/discussions/557\" target=\"_blank\">\ud83d\udcd4 Document</a>": "* \"Image Prompt\" is powered by Fooocus Image Mixture Engine (v1.0.1). <a href=\"https://github.com/lllyasviel/Fooocus/discussions/557\" target=\"_blank\">\ud83d\udcd4 Document</a>",
50
- "Inpaint or Outpaint": "Inpaint or Outpaint",
51
- "Drag inpaint or outpaint image to here": "Drag inpaint or outpaint image to here",
52
- "Mask Upload": "Mask Upload",
53
- "Inpaint Additional Prompt": "Inpaint Additional Prompt",
54
- "Outpaint Direction": "Outpaint Direction",
55
  "Left": "Left",
56
  "Right": "Right",
57
  "Top": "Top",
58
  "Bottom": "Bottom",
59
- "Method": "Method",
60
- "Inpaint or Outpaint (default)": "Inpaint or Outpaint (default)",
61
- "Improve Detail (face, hand, eyes, etc.)": "Improve Detail (face, hand, eyes, etc.)",
62
- "Modify Content (add objects, change background, etc.)": "Modify Content (add objects, change background, etc.)",
63
- "Additional Prompt Quick List": "Additional Prompt Quick List",
64
- "* Powered by Fooocus Inpaint Engine <a href=\"https://github.com/lllyasviel/Fooocus/discussions/414\" target=\"_blank\">\ud83d\udcd4 Document</a>": "* Powered by Fooocus Inpaint Engine <a href=\"https://github.com/lllyasviel/Fooocus/discussions/414\" target=\"_blank\">\ud83d\udcd4 Document</a>",
65
- "Describe": "Describe",
66
- "Drag any image to here": "Drag any image to here",
67
- "Content Type": "Content Type",
68
- "Photograph": "Photograph",
69
- "Art/Anime": "Art/Anime",
70
- "Describe this Image into Prompt": "Describe this Image into Prompt",
71
- "<a href=\"https://github.com/lllyasviel/Fooocus/discussions/1363\" target=\"_blank\">\ud83d\udcd4 Document</a>": "<a href=\"https://github.com/lllyasviel/Fooocus/discussions/1363\" target=\"_blank\">\ud83d\udcd4 Document</a>",
72
- "Metadata": "Metadata",
73
- "Drag any image generated by Fooocus here": "Drag any image generated by Fooocus here",
74
- "Apply Metadata": "Apply Metadata",
75
- "uov": "uov",
76
  "Setting": "Setting",
 
77
  "Preset": "Preset",
78
- "initial": "initial",
79
- "realistic": "realistic",
80
- "lightning": "lightning",
81
- "anime": "anime",
82
- "sai": "sai",
83
- "default": "default",
84
- "lcm": "lcm",
85
  "Performance": "Performance",
86
  "Speed": "Speed",
 
87
  "Extreme Speed": "Extreme Speed",
88
  "Lightning": "Lightning",
89
  "Aspect Ratios": "Aspect Ratios",
90
- "1152\u00d7896 <span style=\"color: grey;\"> \u2223 9:7</span>": "1152\u00d7896 <span style=\"color: grey;\"> \u2223 9:7</span>",
91
  "width \u00d7 height": "width \u00d7 height",
92
- "704\u00d71408 <span style=\"color: grey;\"> \u2223 1:2</span>": "704\u00d71408 <span style=\"color: grey;\"> \u2223 1:2</span>",
93
- "704\u00d71344 <span style=\"color: grey;\"> \u2223 11:21</span>": "704\u00d71344 <span style=\"color: grey;\"> \u2223 11:21</span>",
94
- "768\u00d71344 <span style=\"color: grey;\"> \u2223 4:7</span>": "768\u00d71344 <span style=\"color: grey;\"> \u2223 4:7</span>",
95
- "768\u00d71280 <span style=\"color: grey;\"> \u2223 3:5</span>": "768\u00d71280 <span style=\"color: grey;\"> \u2223 3:5</span>",
96
- "832\u00d71216 <span style=\"color: grey;\"> \u2223 13:19</span>": "832\u00d71216 <span style=\"color: grey;\"> \u2223 13:19</span>",
97
- "832\u00d71152 <span style=\"color: grey;\"> \u2223 13:18</span>": "832\u00d71152 <span style=\"color: grey;\"> \u2223 13:18</span>",
98
- "896\u00d71152 <span style=\"color: grey;\"> \u2223 7:9</span>": "896\u00d71152 <span style=\"color: grey;\"> \u2223 7:9</span>",
99
- "896\u00d71088 <span style=\"color: grey;\"> \u2223 14:17</span>": "896\u00d71088 <span style=\"color: grey;\"> \u2223 14:17</span>",
100
- "960\u00d71088 <span style=\"color: grey;\"> \u2223 15:17</span>": "960\u00d71088 <span style=\"color: grey;\"> \u2223 15:17</span>",
101
- "960\u00d71024 <span style=\"color: grey;\"> \u2223 15:16</span>": "960\u00d71024 <span style=\"color: grey;\"> \u2223 15:16</span>",
102
- "1024\u00d71024 <span style=\"color: grey;\"> \u2223 1:1</span>": "1024\u00d71024 <span style=\"color: grey;\"> \u2223 1:1</span>",
103
- "1024\u00d7960 <span style=\"color: grey;\"> \u2223 16:15</span>": "1024\u00d7960 <span style=\"color: grey;\"> \u2223 16:15</span>",
104
- "1088\u00d7960 <span style=\"color: grey;\"> \u2223 17:15</span>": "1088\u00d7960 <span style=\"color: grey;\"> \u2223 17:15</span>",
105
- "1088\u00d7896 <span style=\"color: grey;\"> \u2223 17:14</span>": "1088\u00d7896 <span style=\"color: grey;\"> \u2223 17:14</span>",
106
- "1152\u00d7832 <span style=\"color: grey;\"> \u2223 18:13</span>": "1152\u00d7832 <span style=\"color: grey;\"> \u2223 18:13</span>",
107
- "1216\u00d7832 <span style=\"color: grey;\"> \u2223 19:13</span>": "1216\u00d7832 <span style=\"color: grey;\"> \u2223 19:13</span>",
108
- "1280\u00d7768 <span style=\"color: grey;\"> \u2223 5:3</span>": "1280\u00d7768 <span style=\"color: grey;\"> \u2223 5:3</span>",
109
- "1344\u00d7768 <span style=\"color: grey;\"> \u2223 7:4</span>": "1344\u00d7768 <span style=\"color: grey;\"> \u2223 7:4</span>",
110
- "1344\u00d7704 <span style=\"color: grey;\"> \u2223 21:11</span>": "1344\u00d7704 <span style=\"color: grey;\"> \u2223 21:11</span>",
111
- "1408\u00d7704 <span style=\"color: grey;\"> \u2223 2:1</span>": "1408\u00d7704 <span style=\"color: grey;\"> \u2223 2:1</span>",
112
- "1472\u00d7704 <span style=\"color: grey;\"> \u2223 23:11</span>": "1472\u00d7704 <span style=\"color: grey;\"> \u2223 23:11</span>",
113
- "1536\u00d7640 <span style=\"color: grey;\"> \u2223 12:5</span>": "1536\u00d7640 <span style=\"color: grey;\"> \u2223 12:5</span>",
114
- "1600\u00d7640 <span style=\"color: grey;\"> \u2223 5:2</span>": "1600\u00d7640 <span style=\"color: grey;\"> \u2223 5:2</span>",
115
- "1664\u00d7576 <span style=\"color: grey;\"> \u2223 26:9</span>": "1664\u00d7576 <span style=\"color: grey;\"> \u2223 26:9</span>",
116
- "1728\u00d7576 <span style=\"color: grey;\"> \u2223 3:1</span>": "1728\u00d7576 <span style=\"color: grey;\"> \u2223 3:1</span>",
117
  "Image Number": "Image Number",
118
- "Output Format": "Output Format",
119
- "png": "png",
120
- "jpeg": "jpeg",
121
- "webp": "webp",
122
  "Negative Prompt": "Negative Prompt",
123
  "Describing what you do not want to see.": "Describing what you do not want to see.",
124
  "Random": "Random",
125
  "Seed": "Seed",
126
- "0": "0",
127
- "Style": "Style",
128
- "Search Styles": "Search Styles",
129
- "Selected Styles": "Selected Styles",
 
130
  "Fooocus V2": "Fooocus V2",
131
- "Fooocus Enhance": "Fooocus Enhance",
132
- "Fooocus Sharp": "Fooocus Sharp",
133
- "Fooocus Semi Realistic": "Fooocus Semi Realistic",
134
  "Fooocus Masterpiece": "Fooocus Masterpiece",
135
  "Fooocus Photograph": "Fooocus Photograph",
136
  "Fooocus Negative": "Fooocus Negative",
137
- "Fooocus Cinematic": "Fooocus Cinematic",
138
  "SAI 3D Model": "SAI 3D Model",
139
  "SAI Analog Film": "SAI Analog Film",
140
  "SAI Anime": "SAI Anime",
@@ -342,90 +268,35 @@
342
  "Volumetric Lighting": "Volumetric Lighting",
343
  "Watercolor 2": "Watercolor 2",
344
  "Whimsical And Playful": "Whimsical And Playful",
345
- "Mk Chromolithography": "Mk Chromolithography",
346
- "Mk Cross Processing Print": "Mk Cross Processing Print",
347
- "Mk Dufaycolor Photograph": "Mk Dufaycolor Photograph",
348
- "Mk Herbarium": "Mk Herbarium",
349
- "Mk Punk Collage": "Mk Punk Collage",
350
- "Mk Mosaic": "Mk Mosaic",
351
- "Mk Van Gogh": "Mk Van Gogh",
352
- "Mk Coloring Book": "Mk Coloring Book",
353
- "Mk Singer Sargent": "Mk Singer Sargent",
354
- "Mk Pollock": "Mk Pollock",
355
- "Mk Basquiat": "Mk Basquiat",
356
- "Mk Andy Warhol": "Mk Andy Warhol",
357
- "Mk Halftone Print": "Mk Halftone Print",
358
- "Mk Gond Painting": "Mk Gond Painting",
359
- "Mk Albumen Print": "Mk Albumen Print",
360
- "Mk Aquatint Print": "Mk Aquatint Print",
361
- "Mk Anthotype Print": "Mk Anthotype Print",
362
- "Mk Inuit Carving": "Mk Inuit Carving",
363
- "Mk Bromoil Print": "Mk Bromoil Print",
364
- "Mk Calotype Print": "Mk Calotype Print",
365
- "Mk Color Sketchnote": "Mk Color Sketchnote",
366
- "Mk Cibulak Porcelain": "Mk Cibulak Porcelain",
367
- "Mk Alcohol Ink Art": "Mk Alcohol Ink Art",
368
- "Mk One Line Art": "Mk One Line Art",
369
- "Mk Blacklight Paint": "Mk Blacklight Paint",
370
- "Mk Carnival Glass": "Mk Carnival Glass",
371
- "Mk Cyanotype Print": "Mk Cyanotype Print",
372
- "Mk Cross Stitching": "Mk Cross Stitching",
373
- "Mk Encaustic Paint": "Mk Encaustic Paint",
374
- "Mk Embroidery": "Mk Embroidery",
375
- "Mk Gyotaku": "Mk Gyotaku",
376
- "Mk Luminogram": "Mk Luminogram",
377
- "Mk Lite Brite Art": "Mk Lite Brite Art",
378
- "Mk Mokume Gane": "Mk Mokume Gane",
379
- "Pebble Art": "Pebble Art",
380
- "Mk Palekh": "Mk Palekh",
381
- "Mk Suminagashi": "Mk Suminagashi",
382
- "Mk Scrimshaw": "Mk Scrimshaw",
383
- "Mk Shibori": "Mk Shibori",
384
- "Mk Vitreous Enamel": "Mk Vitreous Enamel",
385
- "Mk Ukiyo E": "Mk Ukiyo E",
386
- "Mk Vintage Airline Poster": "Mk Vintage Airline Poster",
387
- "Mk Vintage Travel Poster": "Mk Vintage Travel Poster",
388
- "Mk Bauhaus Style": "Mk Bauhaus Style",
389
- "Mk Afrofuturism": "Mk Afrofuturism",
390
- "Mk Atompunk": "Mk Atompunk",
391
- "Mk Constructivism": "Mk Constructivism",
392
- "Mk Chicano Art": "Mk Chicano Art",
393
- "Mk De Stijl": "Mk De Stijl",
394
- "Mk Dayak Art": "Mk Dayak Art",
395
- "Mk Fayum Portrait": "Mk Fayum Portrait",
396
- "Mk Illuminated Manuscript": "Mk Illuminated Manuscript",
397
- "Mk Kalighat Painting": "Mk Kalighat Painting",
398
- "Mk Madhubani Painting": "Mk Madhubani Painting",
399
- "Mk Pictorialism": "Mk Pictorialism",
400
- "Mk Pichwai Painting": "Mk Pichwai Painting",
401
- "Mk Patachitra Painting": "Mk Patachitra Painting",
402
- "Mk Samoan Art Inspired": "Mk Samoan Art Inspired",
403
- "Mk Tlingit Art": "Mk Tlingit Art",
404
- "Mk Adnate Style": "Mk Adnate Style",
405
- "Mk Ron English Style": "Mk Ron English Style",
406
- "Mk Shepard Fairey Style": "Mk Shepard Fairey Style",
407
  "Base Model (SDXL only)": "Base Model (SDXL only)",
408
- "juggernautXL_v8Rundiffusion.safetensors": "juggernautXL_v8Rundiffusion.safetensors",
 
 
 
 
 
 
409
  "Refiner (SDXL or SD 1.5)": "Refiner (SDXL or SD 1.5)",
410
  "None": "None",
411
- "Refiner Switch At": "Refiner Switch At",
412
- "Use 0.4 for SD1.5 realistic models; or 0.667 for SD1.5 anime models; or 0.8 for XL-refiners; or any value for switching two SDXL models.": "Use 0.4 for SD1.5 realistic models; or 0.667 for SD1.5 anime models; or 0.8 for XL-refiners; or any value for switching two SDXL models.",
413
- "Enable": "Enable",
414
- "LoRA 1": "LoRA 1",
415
  "sd_xl_offset_example-lora_1.0.safetensors": "sd_xl_offset_example-lora_1.0.safetensors",
416
- "LoRA 2": "LoRA 2",
417
- "LoRA 3": "LoRA 3",
418
- "LoRA 4": "LoRA 4",
419
- "LoRA 5": "LoRA 5",
 
 
 
420
  "Refresh": "Refresh",
421
  "\ud83d\udd04 Refresh All Files": "\ud83d\udd04 Refresh All Files",
 
 
422
  "Guidance Scale": "Guidance Scale",
423
  "Higher value means style is cleaner, vivider, and more artistic.": "Higher value means style is cleaner, vivider, and more artistic.",
424
- "Image Sharpness": "Image Sharpness",
425
- "Higher value means image and texture are sharper.": "Higher value means image and texture are sharper.",
426
- "<a href=\"https://github.com/lllyasviel/Fooocus/discussions/117\" target=\"_blank\">\ud83d\udcd4 Document</a>": "<a href=\"https://github.com/lllyasviel/Fooocus/discussions/117\" target=\"_blank\">\ud83d\udcd4 Document</a>",
427
  "Developer Debug Mode": "Developer Debug Mode",
428
- "Debug Tools": "Debug Tools",
429
  "Positive ADM Guidance Scaler": "Positive ADM Guidance Scaler",
430
  "The scaler multiplied to positive ADM (use 1.0 to disable). ": "The scaler multiplied to positive ADM (use 1.0 to disable). ",
431
  "Negative ADM Guidance Scaler": "Negative ADM Guidance Scaler",
@@ -440,10 +311,10 @@
440
  "Enabling Fooocus's implementation of CFG mimicking for TSNR (effective when real CFG > mimicked CFG).": "Enabling Fooocus's implementation of CFG mimicking for TSNR (effective when real CFG > mimicked CFG).",
441
  "Sampler": "Sampler",
442
  "dpmpp_2m_sde_gpu": "dpmpp_2m_sde_gpu",
 
443
  "euler": "euler",
444
  "euler_ancestral": "euler_ancestral",
445
  "heun": "heun",
446
- "heunpp2": "heunpp2",
447
  "dpm_2": "dpm_2",
448
  "dpm_2_ancestral": "dpm_2_ancestral",
449
  "lms": "lms",
@@ -462,14 +333,12 @@
462
  "uni_pc_bh2": "uni_pc_bh2",
463
  "Scheduler": "Scheduler",
464
  "karras": "karras",
 
465
  "normal": "normal",
466
  "exponential": "exponential",
467
  "sgm_uniform": "sgm_uniform",
468
  "simple": "simple",
469
  "ddim_uniform": "ddim_uniform",
470
- "turbo": "turbo",
471
- "Generate Image Grid for Each Batch": "Generate Image Grid for Each Batch",
472
- "(Experimental) This may cause performance problems on some computers and certain internet conditions.": "(Experimental) This may cause performance problems on some computers and certain internet conditions.",
473
  "Forced Overwrite of Sampling Step": "Forced Overwrite of Sampling Step",
474
  "Set as -1 to disable. For developer debugging.": "Set as -1 to disable. For developer debugging.",
475
  "Forced Overwrite of Refiner Switch Step": "Forced Overwrite of Refiner Switch Step",
@@ -483,22 +352,12 @@
483
  "Disable preview during generation.": "Disable preview during generation.",
484
  "Disable Intermediate Results": "Disable Intermediate Results",
485
  "Disable intermediate results during generation, only show final gallery.": "Disable intermediate results during generation, only show final gallery.",
486
- "Disable seed increment": "Disable seed increment",
487
- "Disable automatic seed increment when image number is > 1.": "Disable automatic seed increment when image number is > 1.",
488
- "Read wildcards in order": "Read wildcards in order",
489
- "Save Metadata to Images": "Save Metadata to Images",
490
- "Adds parameters to generated images allowing manual regeneration.": "Adds parameters to generated images allowing manual regeneration.",
491
- "Metadata Scheme": "Metadata Scheme",
492
- "fooocus": "fooocus",
493
- "Image Prompt parameters are not included. Use png and a1111 for compatibility with Civitai.": "Image Prompt parameters are not included. Use png and a1111 for compatibility with Civitai.",
494
- "fooocus (json)": "fooocus (json)",
495
- "a1111 (plain text)": "a1111 (plain text)",
496
- "a1111": "a1111",
497
- "Control": "Control",
498
  "Debug Preprocessors": "Debug Preprocessors",
499
- "See the results from preprocessors.": "See the results from preprocessors.",
500
- "Skip Preprocessors": "Skip Preprocessors",
501
- "Do not preprocess images. (Inputs are already canny/depth/cropped-face/etc.)": "Do not preprocess images. (Inputs are already canny/depth/cropped-face/etc.)",
502
  "Mixing Image Prompt and Vary/Upscale": "Mixing Image Prompt and Vary/Upscale",
503
  "Mixing Image Prompt and Inpaint": "Mixing Image Prompt and Inpaint",
504
  "Softness of ControlNet": "Softness of ControlNet",
@@ -506,26 +365,25 @@
506
  "Canny": "Canny",
507
  "Canny Low Threshold": "Canny Low Threshold",
508
  "Canny High Threshold": "Canny High Threshold",
509
- "Inpaint": "Inpaint",
510
- "Debug Inpaint Preprocessing": "Debug Inpaint Preprocessing",
511
- "Disable initial latent in inpaint": "Disable initial latent in inpaint",
512
- "Inpaint Engine": "Inpaint Engine",
513
- "v2.6": "v2.6",
514
- "Version of Fooocus inpaint model": "Version of Fooocus inpaint model",
515
- "v1": "v1",
516
- "v2.5": "v2.5",
517
- "Inpaint Denoising Strength": "Inpaint Denoising Strength",
518
- "Same as the denoising strength in A1111 inpaint. Only used in inpaint, not used in outpaint. (Outpaint always use 1.0)": "Same as the denoising strength in A1111 inpaint. Only used in inpaint, not used in outpaint. (Outpaint always use 1.0)",
519
- "Inpaint Respective Field": "Inpaint Respective Field",
520
- "The area to inpaint. Value 0 is same as \"Only Masked\" in A1111. Value 1 is same as \"Whole Image\" in A1111. Only used in inpaint, not used in outpaint. (Outpaint always use 1.0)": "The area to inpaint. Value 0 is same as \"Only Masked\" in A1111. Value 1 is same as \"Whole Image\" in A1111. Only used in inpaint, not used in outpaint. (Outpaint always use 1.0)",
521
- "Mask Erode or Dilate": "Mask Erode or Dilate",
522
- "Positive value will make white area in the mask larger, negative value will make white area smaller.(default is 0, always process before any mask invert)": "Positive value will make white area in the mask larger, negative value will make white area smaller.(default is 0, always process before any mask invert)",
523
- "Enable Mask Upload": "Enable Mask Upload",
524
- "Invert Mask": "Invert Mask",
525
  "FreeU": "FreeU",
526
  "Enabled": "Enabled",
527
  "B1": "B1",
528
  "B2": "B2",
529
  "S1": "S1",
530
- "S2": "S2"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
531
  }
 
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "Preview": "Preview",
 
3
  "Gallery": "Gallery",
4
  "Generate": "Generate",
 
5
  "Skip": "Skip",
6
  "Stop": "Stop",
7
  "Input Image": "Input Image",
8
  "Advanced": "Advanced",
9
  "Upscale or Variation": "Upscale or Variation",
10
+ "Image Prompt": "Image Prompt",
11
+ "Inpaint or Outpaint (beta)": "Inpaint or Outpaint (beta)",
12
  "Drag above image to here": "Drag above image to here",
13
  "Upscale or Variation:": "Upscale or Variation:",
14
  "Disabled": "Disabled",
 
17
  "Upscale (1.5x)": "Upscale (1.5x)",
18
  "Upscale (2x)": "Upscale (2x)",
19
  "Upscale (Fast 2x)": "Upscale (Fast 2x)",
20
+ "\ud83d\udcd4 Document": "\uD83D\uDCD4 Document",
 
21
  "Image": "Image",
22
  "Stop At": "Stop At",
23
  "Weight": "Weight",
24
  "Type": "Type",
 
25
  "PyraCanny": "PyraCanny",
26
  "CPDS": "CPDS",
27
+ "* \"Image Prompt\" is powered by Fooocus Image Mixture Engine (v1.0.1).": "* \"Image Prompt\" is powered by Fooocus Image Mixture Engine (v1.0.1).",
28
+ "The scaler multiplied to positive ADM (use 1.0 to disable).": "The scaler multiplied to positive ADM (use 1.0 to disable).",
29
+ "The scaler multiplied to negative ADM (use 1.0 to disable).": "The scaler multiplied to negative ADM (use 1.0 to disable).",
30
+ "When to end the guidance from positive/negative ADM.": "When to end the guidance from positive/negative ADM.",
31
+ "Similar to the Control Mode in A1111 (use 0.0 to disable).": "Similar to the Control Mode in A1111 (use 0.0 to disable).",
32
+ "Outpaint Expansion (": "Outpaint Expansion (",
33
+ "Outpaint": "Outpaint",
34
  "Left": "Left",
35
  "Right": "Right",
36
  "Top": "Top",
37
  "Bottom": "Bottom",
38
+ "* \"Inpaint or Outpaint\" is powered by the sampler \"DPMPP Fooocus Seamless 2M SDE Karras Inpaint Sampler\" (beta)": "* \"Inpaint or Outpaint\" is powered by the sampler \"DPMPP Fooocus Seamless 2M SDE Karras Inpaint Sampler\" (beta)",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
  "Setting": "Setting",
40
+ "Style": "Style",
41
  "Preset": "Preset",
 
 
 
 
 
 
 
42
  "Performance": "Performance",
43
  "Speed": "Speed",
44
+ "Quality": "Quality",
45
  "Extreme Speed": "Extreme Speed",
46
  "Lightning": "Lightning",
47
  "Aspect Ratios": "Aspect Ratios",
 
48
  "width \u00d7 height": "width \u00d7 height",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
49
  "Image Number": "Image Number",
 
 
 
 
50
  "Negative Prompt": "Negative Prompt",
51
  "Describing what you do not want to see.": "Describing what you do not want to see.",
52
  "Random": "Random",
53
  "Seed": "Seed",
54
+ "Disable seed increment": "Disable seed increment",
55
+ "Disable automatic seed increment when image number is > 1.": "Disable automatic seed increment when image number is > 1.",
56
+ "Read wildcards in order": "Read wildcards in order",
57
+ "\ud83d\udcda History Log": "\uD83D\uDCDA History Log",
58
+ "Image Style": "Image Style",
59
  "Fooocus V2": "Fooocus V2",
60
+ "Default (Slightly Cinematic)": "Default (Slightly Cinematic)",
 
 
61
  "Fooocus Masterpiece": "Fooocus Masterpiece",
62
  "Fooocus Photograph": "Fooocus Photograph",
63
  "Fooocus Negative": "Fooocus Negative",
 
64
  "SAI 3D Model": "SAI 3D Model",
65
  "SAI Analog Film": "SAI Analog Film",
66
  "SAI Anime": "SAI Anime",
 
268
  "Volumetric Lighting": "Volumetric Lighting",
269
  "Watercolor 2": "Watercolor 2",
270
  "Whimsical And Playful": "Whimsical And Playful",
271
+ "Model": "Model",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
272
  "Base Model (SDXL only)": "Base Model (SDXL only)",
273
+ "sd_xl_base_1.0_0.9vae.safetensors": "sd_xl_base_1.0_0.9vae.safetensors",
274
+ "bluePencilXL_v009.safetensors": "bluePencilXL_v009.safetensors",
275
+ "bluePencilXL_v050.safetensors": "bluePencilXL_v050.safetensors",
276
+ "DreamShaper_8_pruned.safetensors": "DreamShaper_8_pruned.safetensors",
277
+ "realisticStockPhoto_v10.safetensors": "realisticStockPhoto_v10.safetensors",
278
+ "realisticVisionV51_v51VAE.safetensors": "realisticVisionV51_v51VAE.safetensors",
279
+ "sd_xl_refiner_1.0_0.9vae.safetensors": "sd_xl_refiner_1.0_0.9vae.safetensors",
280
  "Refiner (SDXL or SD 1.5)": "Refiner (SDXL or SD 1.5)",
281
  "None": "None",
282
+ "LoRAs": "LoRAs",
283
+ "SDXL LoRA 1": "SDXL LoRA 1",
 
 
284
  "sd_xl_offset_example-lora_1.0.safetensors": "sd_xl_offset_example-lora_1.0.safetensors",
285
+ "3d_render_style_xl.safetensors": "3d_render_style_xl.safetensors",
286
+ "Bloodstained-XL-V1.safetensors": "Bloodstained-XL-V1.safetensors",
287
+ "SDXL_FILM_PHOTOGRAPHY_STYLE_BetaV0.4.safetensors": "SDXL_FILM_PHOTOGRAPHY_STYLE_BetaV0.4.safetensors",
288
+ "SDXL LoRA 2": "SDXL LoRA 2",
289
+ "SDXL LoRA 3": "SDXL LoRA 3",
290
+ "SDXL LoRA 4": "SDXL LoRA 4",
291
+ "SDXL LoRA 5": "SDXL LoRA 5",
292
  "Refresh": "Refresh",
293
  "\ud83d\udd04 Refresh All Files": "\ud83d\udd04 Refresh All Files",
294
+ "Sampling Sharpness": "Sampling Sharpness",
295
+ "Higher value means image and texture are sharper.": "Higher value means image and texture are sharper.",
296
  "Guidance Scale": "Guidance Scale",
297
  "Higher value means style is cleaner, vivider, and more artistic.": "Higher value means style is cleaner, vivider, and more artistic.",
 
 
 
298
  "Developer Debug Mode": "Developer Debug Mode",
299
+ "Developer Debug Tools": "Developer Debug Tools",
300
  "Positive ADM Guidance Scaler": "Positive ADM Guidance Scaler",
301
  "The scaler multiplied to positive ADM (use 1.0 to disable). ": "The scaler multiplied to positive ADM (use 1.0 to disable). ",
302
  "Negative ADM Guidance Scaler": "Negative ADM Guidance Scaler",
 
311
  "Enabling Fooocus's implementation of CFG mimicking for TSNR (effective when real CFG > mimicked CFG).": "Enabling Fooocus's implementation of CFG mimicking for TSNR (effective when real CFG > mimicked CFG).",
312
  "Sampler": "Sampler",
313
  "dpmpp_2m_sde_gpu": "dpmpp_2m_sde_gpu",
314
+ "Only effective in non-inpaint mode.": "Only effective in non-inpaint mode.",
315
  "euler": "euler",
316
  "euler_ancestral": "euler_ancestral",
317
  "heun": "heun",
 
318
  "dpm_2": "dpm_2",
319
  "dpm_2_ancestral": "dpm_2_ancestral",
320
  "lms": "lms",
 
333
  "uni_pc_bh2": "uni_pc_bh2",
334
  "Scheduler": "Scheduler",
335
  "karras": "karras",
336
+ "Scheduler of Sampler.": "Scheduler of Sampler.",
337
  "normal": "normal",
338
  "exponential": "exponential",
339
  "sgm_uniform": "sgm_uniform",
340
  "simple": "simple",
341
  "ddim_uniform": "ddim_uniform",
 
 
 
342
  "Forced Overwrite of Sampling Step": "Forced Overwrite of Sampling Step",
343
  "Set as -1 to disable. For developer debugging.": "Set as -1 to disable. For developer debugging.",
344
  "Forced Overwrite of Refiner Switch Step": "Forced Overwrite of Refiner Switch Step",
 
352
  "Disable preview during generation.": "Disable preview during generation.",
353
  "Disable Intermediate Results": "Disable Intermediate Results",
354
  "Disable intermediate results during generation, only show final gallery.": "Disable intermediate results during generation, only show final gallery.",
355
+ "Inpaint Engine": "Inpaint Engine",
356
+ "v1": "v1",
357
+ "Version of Fooocus inpaint model": "Version of Fooocus inpaint model",
358
+ "v2.5": "v2.5",
359
+ "Control Debug": "Control Debug",
 
 
 
 
 
 
 
360
  "Debug Preprocessors": "Debug Preprocessors",
 
 
 
361
  "Mixing Image Prompt and Vary/Upscale": "Mixing Image Prompt and Vary/Upscale",
362
  "Mixing Image Prompt and Inpaint": "Mixing Image Prompt and Inpaint",
363
  "Softness of ControlNet": "Softness of ControlNet",
 
365
  "Canny": "Canny",
366
  "Canny Low Threshold": "Canny Low Threshold",
367
  "Canny High Threshold": "Canny High Threshold",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
368
  "FreeU": "FreeU",
369
  "Enabled": "Enabled",
370
  "B1": "B1",
371
  "B2": "B2",
372
  "S1": "S1",
373
+ "S2": "S2",
374
+ "\uD83D\uDD0E Type here to search styles ...": "\uD83D\uDD0E Type here to search styles ...",
375
+ "Type prompt here.": "Type prompt here.",
376
+ "Outpaint Expansion Direction:": "Outpaint Expansion Direction:",
377
+ "* Powered by Fooocus Inpaint Engine (beta)": "* Powered by Fooocus Inpaint Engine (beta)",
378
+ "Fooocus Enhance": "Fooocus Enhance",
379
+ "Fooocus Cinematic": "Fooocus Cinematic",
380
+ "Fooocus Sharp": "Fooocus Sharp",
381
+ "Drag any image generated by Fooocus here": "Drag any image generated by Fooocus here",
382
+ "Metadata": "Metadata",
383
+ "Apply Metadata": "Apply Metadata",
384
+ "Metadata Scheme": "Metadata Scheme",
385
+ "Image Prompt parameters are not included. Use png and a1111 for compatibility with Civitai.": "Image Prompt parameters are not included. Use png and a1111 for compatibility with Civitai.",
386
+ "fooocus (json)": "fooocus (json)",
387
+ "a1111 (plain text)": "a1111 (plain text)",
388
+ "Unsupported image type in input": "Unsupported image type in input"
389
  }
launch.py CHANGED
@@ -27,7 +27,6 @@ TRY_INSTALL_XFORMERS = False
27
 
28
 
29
  def prepare_environment():
30
-
31
  torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu121")
32
  torch_command = os.environ.get('TORCH_COMMAND',
33
  f"pip install torch==2.1.0 torchvision==0.16.0 --extra-index-url {torch_index_url}")
@@ -72,12 +71,10 @@ def ini_args():
72
  from args_manager import args
73
  return args
74
 
75
- args = ini_args()
76
- if not args.api_mode:
77
- prepare_environment()
78
- build_launcher()
79
- args = ini_args()
80
 
 
 
 
81
 
82
  if args.gpu_device_id is not None:
83
  os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_device_id)
@@ -131,9 +128,9 @@ def download_models(default_model, previous_default_models, checkpoint_downloads
131
 
132
  return default_model, checkpoint_downloads
133
 
134
- if not args.api_mode:
135
- config.default_base_model_name, config.checkpoint_downloads = download_models(
136
- config.default_base_model_name, config.previous_default_models, config.checkpoint_downloads,
137
- config.embeddings_downloads, config.lora_downloads)
138
 
139
  from webui import *
 
27
 
28
 
29
  def prepare_environment():
 
30
  torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu121")
31
  torch_command = os.environ.get('TORCH_COMMAND',
32
  f"pip install torch==2.1.0 torchvision==0.16.0 --extra-index-url {torch_index_url}")
 
71
  from args_manager import args
72
  return args
73
 
 
 
 
 
 
74
 
75
+ prepare_environment()
76
+ build_launcher()
77
+ args = ini_args()
78
 
79
  if args.gpu_device_id is not None:
80
  os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_device_id)
 
128
 
129
  return default_model, checkpoint_downloads
130
 
131
+
132
+ config.default_base_model_name, config.checkpoint_downloads = download_models(
133
+ config.default_base_model_name, config.previous_default_models, config.checkpoint_downloads,
134
+ config.embeddings_downloads, config.lora_downloads)
135
 
136
  from webui import *
ldm_patched/modules/args_parser.py CHANGED
@@ -46,7 +46,6 @@ parser.add_argument("--in-browser", action="store_true")
46
  parser.add_argument("--disable-in-browser", action="store_true")
47
  parser.add_argument("--gpu-device-id", type=int, default=None, metavar="DEVICE_ID")
48
  cm_group = parser.add_mutually_exclusive_group()
49
- parser.add_argument("--api-mode", action="store_true")
50
  cm_group.add_argument("--async-cuda-allocation", action="store_true")
51
  cm_group.add_argument("--disable-async-cuda-allocation", action="store_true")
52
 
 
46
  parser.add_argument("--disable-in-browser", action="store_true")
47
  parser.add_argument("--gpu-device-id", type=int, default=None, metavar="DEVICE_ID")
48
  cm_group = parser.add_mutually_exclusive_group()
 
49
  cm_group.add_argument("--async-cuda-allocation", action="store_true")
50
  cm_group.add_argument("--disable-async-cuda-allocation", action="store_true")
51
 
modules/ui_gradio_extensions.py CHANGED
@@ -52,11 +52,6 @@ def css_html():
52
  head = f'<link rel="stylesheet" property="stylesheet" href="{style_css_path}">'
53
  return head
54
 
55
- def zStudiosHeader(power_device):
56
- logo_path = webpath('backstage.png')
57
- zHead = f"""<table><tr><td><img src="{logo_path}" width="36" height="36"/></td><td><font size="5">zStudios Generative Media AI running on {power_device}</font></td></tr></table>"""
58
- return zHead
59
-
60
 
61
  def reload_javascript():
62
  js = javascript_html()
 
52
  head = f'<link rel="stylesheet" property="stylesheet" href="{style_css_path}">'
53
  return head
54
 
 
 
 
 
 
55
 
56
  def reload_javascript():
57
  js = javascript_html()
requirements.txt CHANGED
@@ -20,4 +20,3 @@ timm==0.9.2
20
  diffusers
21
  invisible_watermark
22
  xformers
23
- openai==1.2.3
 
20
  diffusers
21
  invisible_watermark
22
  xformers
 
webui.py CHANGED
@@ -16,7 +16,6 @@ import modules.meta_parser
16
  import args_manager
17
  import copy
18
  import launch
19
- import torch
20
 
21
  from modules.sdxl_styles import legal_style_names
22
  from modules.private_logger import get_current_html_path
@@ -24,15 +23,6 @@ from modules.ui_gradio_extensions import reload_javascript
24
  from modules.auth import auth_enabled, check_auth
25
  from modules.util import is_json
26
 
27
- def ini_args():
28
- from args_manager import args
29
- return args
30
-
31
- if ini_args().api_mode:
32
- import dummy_inference as inf
33
- else:
34
- import inference as inf
35
-
36
  def get_task(*args):
37
  args = list(args)
38
  args.pop(0)
@@ -106,642 +96,614 @@ title = f'Fooocus {fooocus_version.version}'
106
  if isinstance(args_manager.args.preset, str):
107
  title += ' ' + args_manager.args.preset
108
 
109
- theme = gr.themes.Soft()
110
-
111
- shared.gradio_root = gr.Blocks(title=title, theme=theme).queue()
112
-
113
- if torch.cuda.is_available():
114
- power_device = "GPU"
115
- else:
116
- power_device = "CPU"
117
 
118
  with shared.gradio_root:
119
- gr.Markdown(f"""# <center>zStudios Generative Media AI running on {power_device}</center>""")
120
- with gr.Tab("DALL-E"):
121
- with gr.Row(variant="panel"):
122
- model = gr.Dropdown(choices=["dall-e-2", "dall-e-3"], label="Model", value="dall-e-3")
123
- quality = gr.Dropdown(choices=["standard", "hd"], label="Quality", value="standard")
124
- size = gr.Dropdown(choices=["256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"], label="Size",
125
- value="1024x1024")
126
-
127
- dalleText = gr.Textbox(label="Prompt",
128
- placeholder="Enter your text and then click on the \"Run\" button, "
129
- "or simply press the Enter key.")
130
- dalleBtn = gr.Button("Run")
131
- output_image = gr.Image(label="Image Output")
132
-
133
- dalleText.submit(fn=inf.infer_dall_e, inputs=[dalleText, model, quality, size], outputs=output_image, api_name="infer_dall_e")
134
- dalleBtn.click(fn=inf.infer_dall_e, inputs=[dalleText, model, quality, size], outputs=output_image, api_name=False)
135
-
136
- with gr.Tab("Fooocus"):
137
- args = ini_args()
138
- if args.api_mode:
139
- gr.Markdown("Deactivated")
140
- currentTask = gr.State(worker.AsyncTask(args=[]))
141
- with gr.Row():
142
- with gr.Column(scale=2):
143
- with gr.Row():
144
- progress_window = grh.Image(label='Preview', show_label=True, visible=False, height=768,
145
- elem_classes=['main_view'])
146
- progress_gallery = gr.Gallery(label='Finished Images', show_label=True, object_fit='contain',
147
- height=768, visible=False, elem_classes=['main_view', 'image_gallery'])
148
- progress_html = gr.HTML(value=modules.html.make_progress_html(32, 'Progress 32%'), visible=False,
149
- elem_id='progress-bar', elem_classes='progress-bar')
150
- gallery = gr.Gallery(label='Gallery', show_label=False, object_fit='contain', visible=True, height=768,
151
- elem_classes=['resizable_area', 'main_view', 'final_gallery', 'image_gallery'],
152
- elem_id='final_gallery')
153
- with gr.Row(elem_classes='type_row'):
154
- with gr.Column(scale=17):
155
- prompt = gr.Textbox(show_label=False, placeholder="Type prompt here or paste parameters.", elem_id='positive_prompt',
156
- container=False, autofocus=True, elem_classes='type_row', lines=1024)
157
-
158
- default_prompt = modules.config.default_prompt
159
- if isinstance(default_prompt, str) and default_prompt != '':
160
- shared.gradio_root.load(lambda: default_prompt, outputs=prompt)
161
-
162
- with gr.Column(scale=3, min_width=0):
163
- generate_button = gr.Button(label="Generate", value="Generate", elem_classes='type_row', elem_id='generate_button', visible=True)
164
- load_parameter_button = gr.Button(label="Load Parameters", value="Load Parameters", elem_classes='type_row', elem_id='load_parameter_button', visible=False)
165
- skip_button = gr.Button(label="Skip", value="Skip", elem_classes='type_row_half', visible=False)
166
- stop_button = gr.Button(label="Stop", value="Stop", elem_classes='type_row_half', elem_id='stop_button', visible=False)
167
-
168
- def stop_clicked(currentTask):
169
- import ldm_patched.modules.model_management as model_management
170
- currentTask.last_stop = 'stop'
171
- if (currentTask.processing):
172
- model_management.interrupt_current_processing()
173
- return currentTask
174
-
175
- def skip_clicked(currentTask):
176
- import ldm_patched.modules.model_management as model_management
177
- currentTask.last_stop = 'skip'
178
- if (currentTask.processing):
179
- model_management.interrupt_current_processing()
180
- return currentTask
181
-
182
- stop_button.click(stop_clicked, inputs=currentTask, outputs=currentTask, queue=False, show_progress=False, _js='cancelGenerateForever')
183
- skip_button.click(skip_clicked, inputs=currentTask, outputs=currentTask, queue=False, show_progress=False)
184
- with gr.Row(elem_classes='advanced_check_row'):
185
- input_image_checkbox = gr.Checkbox(label='Input Image', value=False, container=False, elem_classes='min_check')
186
- advanced_checkbox = gr.Checkbox(label='Advanced', value=modules.config.default_advanced_checkbox, container=False, elem_classes='min_check')
187
- with gr.Row(visible=False) as image_input_panel:
188
- with gr.Tabs():
189
- with gr.TabItem(label='Upscale or Variation') as uov_tab:
190
- with gr.Row():
191
- with gr.Column():
192
- uov_input_image = grh.Image(label='Drag above image to here', source='upload', type='numpy')
193
- with gr.Column():
194
- uov_method = gr.Radio(label='Upscale or Variation:', choices=flags.uov_list, value=flags.disabled)
195
- gr.HTML('<a href="https://github.com/lllyasviel/Fooocus/discussions/390" target="_blank">\U0001F4D4 Document</a>')
196
- with gr.TabItem(label='Image Prompt') as ip_tab:
197
- with gr.Row():
198
- ip_images = []
199
- ip_types = []
200
- ip_stops = []
201
- ip_weights = []
202
- ip_ctrls = []
203
- ip_ad_cols = []
204
- for _ in range(flags.controlnet_image_count):
205
- with gr.Column():
206
- ip_image = grh.Image(label='Image', source='upload', type='numpy', show_label=False, height=300)
207
- ip_images.append(ip_image)
208
- ip_ctrls.append(ip_image)
209
- with gr.Column(visible=False) as ad_col:
210
- with gr.Row():
211
- default_end, default_weight = flags.default_parameters[flags.default_ip]
212
-
213
- ip_stop = gr.Slider(label='Stop At', minimum=0.0, maximum=1.0, step=0.001, value=default_end)
214
- ip_stops.append(ip_stop)
215
- ip_ctrls.append(ip_stop)
216
-
217
- ip_weight = gr.Slider(label='Weight', minimum=0.0, maximum=2.0, step=0.001, value=default_weight)
218
- ip_weights.append(ip_weight)
219
- ip_ctrls.append(ip_weight)
220
-
221
- ip_type = gr.Radio(label='Type', choices=flags.ip_list, value=flags.default_ip, container=False)
222
- ip_types.append(ip_type)
223
- ip_ctrls.append(ip_type)
224
-
225
- ip_type.change(lambda x: flags.default_parameters[x], inputs=[ip_type], outputs=[ip_stop, ip_weight], queue=False, show_progress=False)
226
- ip_ad_cols.append(ad_col)
227
- ip_advanced = gr.Checkbox(label='Advanced', value=False, container=False)
228
- gr.HTML('* \"Image Prompt\" is powered by Fooocus Image Mixture Engine (v1.0.1). <a href="https://github.com/lllyasviel/Fooocus/discussions/557" target="_blank">\U0001F4D4 Document</a>')
229
-
230
- def ip_advance_checked(x):
231
- return [gr.update(visible=x)] * len(ip_ad_cols) + \
232
- [flags.default_ip] * len(ip_types) + \
233
- [flags.default_parameters[flags.default_ip][0]] * len(ip_stops) + \
234
- [flags.default_parameters[flags.default_ip][1]] * len(ip_weights)
235
-
236
- ip_advanced.change(ip_advance_checked, inputs=ip_advanced,
237
- outputs=ip_ad_cols + ip_types + ip_stops + ip_weights,
238
- queue=False, show_progress=False)
239
- with gr.TabItem(label='Inpaint or Outpaint') as inpaint_tab:
240
- with gr.Row():
241
- inpaint_input_image = grh.Image(label='Drag inpaint or outpaint image to here', source='upload', type='numpy', tool='sketch', height=500, brush_color="#FFFFFF", elem_id='inpaint_canvas')
242
- inpaint_mask_image = grh.Image(label='Mask Upload', source='upload', type='numpy', height=500, visible=False)
243
-
244
- with gr.Row():
245
- inpaint_additional_prompt = gr.Textbox(placeholder="Describe what you want to inpaint.", elem_id='inpaint_additional_prompt', label='Inpaint Additional Prompt', visible=False)
246
- outpaint_selections = gr.CheckboxGroup(choices=['Left', 'Right', 'Top', 'Bottom'], value=[], label='Outpaint Direction')
247
- inpaint_mode = gr.Dropdown(choices=modules.flags.inpaint_options, value=modules.flags.inpaint_option_default, label='Method')
248
- example_inpaint_prompts = gr.Dataset(samples=modules.config.example_inpaint_prompts, label='Additional Prompt Quick List', components=[inpaint_additional_prompt], visible=False)
249
- gr.HTML('* Powered by Fooocus Inpaint Engine <a href="https://github.com/lllyasviel/Fooocus/discussions/414" target="_blank">\U0001F4D4 Document</a>')
250
- example_inpaint_prompts.click(lambda x: x[0], inputs=example_inpaint_prompts, outputs=inpaint_additional_prompt, show_progress=False, queue=False)
251
- with gr.TabItem(label='Describe') as desc_tab:
252
- with gr.Row():
253
- with gr.Column():
254
- desc_input_image = grh.Image(label='Drag any image to here', source='upload', type='numpy')
255
- with gr.Column():
256
- desc_method = gr.Radio(
257
- label='Content Type',
258
- choices=[flags.desc_type_photo, flags.desc_type_anime],
259
- value=flags.desc_type_photo)
260
- desc_btn = gr.Button(value='Describe this Image into Prompt')
261
- gr.HTML('<a href="https://github.com/lllyasviel/Fooocus/discussions/1363" target="_blank">\U0001F4D4 Document</a>')
262
- with gr.TabItem(label='Metadata') as load_tab:
263
  with gr.Column():
264
- metadata_input_image = grh.Image(label='Drag any image generated by Fooocus here', source='upload', type='filepath')
265
- metadata_json = gr.JSON(label='Metadata')
266
- metadata_import_button = gr.Button(value='Apply Metadata')
267
-
268
- def trigger_metadata_preview(filepath):
269
- parameters, metadata_scheme = modules.meta_parser.read_info_from_image(filepath)
270
-
271
- results = {}
272
- if parameters is not None:
273
- results['parameters'] = parameters
274
-
275
- if isinstance(metadata_scheme, flags.MetadataScheme):
276
- results['metadata_scheme'] = metadata_scheme.value
277
-
278
- return results
279
-
280
- metadata_input_image.upload(trigger_metadata_preview, inputs=metadata_input_image,
281
- outputs=metadata_json, queue=False, show_progress=True)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
282
 
283
- switch_js = "(x) => {if(x){viewer_to_bottom(100);viewer_to_bottom(500);}else{viewer_to_top();} return x;}"
284
- down_js = "() => {viewer_to_bottom();}"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
285
 
286
- input_image_checkbox.change(lambda x: gr.update(visible=x), inputs=input_image_checkbox,
287
- outputs=image_input_panel, queue=False, show_progress=False, _js=switch_js)
288
- ip_advanced.change(lambda: None, queue=False, show_progress=False, _js=down_js)
 
 
 
 
289
 
290
- current_tab = gr.Textbox(value='uov', visible=False)
291
- uov_tab.select(lambda: 'uov', outputs=current_tab, queue=False, _js=down_js, show_progress=False)
292
- inpaint_tab.select(lambda: 'inpaint', outputs=current_tab, queue=False, _js=down_js, show_progress=False)
293
- ip_tab.select(lambda: 'ip', outputs=current_tab, queue=False, _js=down_js, show_progress=False)
294
- desc_tab.select(lambda: 'desc', outputs=current_tab, queue=False, _js=down_js, show_progress=False)
295
 
296
- with gr.Column(scale=1, visible=modules.config.default_advanced_checkbox) as advanced_column:
297
- with gr.Tab(label='Setting'):
298
- if not args_manager.args.disable_preset_selection:
299
- preset_selection = gr.Radio(label='Preset',
300
- choices=modules.config.available_presets,
301
- value=args_manager.args.preset if args_manager.args.preset else "initial",
302
- interactive=True)
303
- performance_selection = gr.Radio(label='Performance',
304
- choices=flags.Performance.list(),
305
- value=modules.config.default_performance)
306
- aspect_ratios_selection = gr.Radio(label='Aspect Ratios', choices=modules.config.available_aspect_ratios,
307
- value=modules.config.default_aspect_ratio, info='width × height',
308
- elem_classes='aspect_ratios')
309
- image_number = gr.Slider(label='Image Number', minimum=1, maximum=modules.config.default_max_image_number, step=1, value=modules.config.default_image_number)
310
-
311
- output_format = gr.Radio(label='Output Format',
312
- choices=flags.OutputFormat.list(),
313
- value=modules.config.default_output_format)
314
-
315
- negative_prompt = gr.Textbox(label='Negative Prompt', show_label=True, placeholder="Type prompt here.",
316
- info='Describing what you do not want to see.', lines=2,
317
- elem_id='negative_prompt',
318
- value=modules.config.default_prompt_negative)
319
- seed_random = gr.Checkbox(label='Random', value=True)
320
- image_seed = gr.Textbox(label='Seed', value=0, max_lines=1, visible=False) # workaround for https://github.com/gradio-app/gradio/issues/5354
321
-
322
- def random_checked(r):
323
- return gr.update(visible=not r)
324
-
325
- def refresh_seed(r, seed_string):
326
- if r:
327
- return random.randint(constants.MIN_SEED, constants.MAX_SEED)
328
- else:
329
- try:
330
- seed_value = int(seed_string)
331
- if constants.MIN_SEED <= seed_value <= constants.MAX_SEED:
332
- return seed_value
333
- except ValueError:
334
- pass
335
- return random.randint(constants.MIN_SEED, constants.MAX_SEED)
336
-
337
- seed_random.change(random_checked, inputs=[seed_random], outputs=[image_seed],
338
- queue=False, show_progress=False)
339
 
340
- def update_history_link():
341
- if args_manager.args.disable_image_log:
342
- return gr.update(value='')
343
-
344
- return gr.update(value=f'<a href="file={get_current_html_path(output_format)}" target="_blank">\U0001F4DA History Log</a>')
345
-
346
- history_link = gr.HTML()
347
- shared.gradio_root.load(update_history_link, outputs=history_link, queue=False, show_progress=False)
348
-
349
- with gr.Tab(label='Style', elem_classes=['style_selections_tab']):
350
- style_sorter.try_load_sorted_styles(
351
- style_names=legal_style_names,
352
- default_selected=modules.config.default_styles)
353
-
354
- style_search_bar = gr.Textbox(show_label=False, container=False,
355
- placeholder="\U0001F50E Type here to search styles ...",
356
- value="",
357
- label='Search Styles')
358
- style_selections = gr.CheckboxGroup(show_label=False, container=False,
359
- choices=copy.deepcopy(style_sorter.all_styles),
360
- value=copy.deepcopy(modules.config.default_styles),
361
- label='Selected Styles',
362
- elem_classes=['style_selections'])
363
- gradio_receiver_style_selections = gr.Textbox(elem_id='gradio_receiver_style_selections', visible=False)
364
-
365
- shared.gradio_root.load(lambda: gr.update(choices=copy.deepcopy(style_sorter.all_styles)),
366
- outputs=style_selections)
367
-
368
- style_search_bar.change(style_sorter.search_styles,
369
- inputs=[style_selections, style_search_bar],
370
- outputs=style_selections,
371
- queue=False,
372
- show_progress=False).then(
373
- lambda: None, _js='()=>{refresh_style_localization();}')
374
-
375
- gradio_receiver_style_selections.input(style_sorter.sort_styles,
376
- inputs=style_selections,
377
- outputs=style_selections,
378
- queue=False,
379
- show_progress=False).then(
380
- lambda: None, _js='()=>{refresh_style_localization();}')
381
-
382
- with gr.Tab(label='Model'):
383
- with gr.Group():
384
  with gr.Row():
385
- base_model = gr.Dropdown(label='Base Model (SDXL only)', choices=modules.config.model_filenames, value=modules.config.default_base_model_name, show_label=True)
386
- refiner_model = gr.Dropdown(label='Refiner (SDXL or SD 1.5)', choices=['None'] + modules.config.model_filenames, value=modules.config.default_refiner_model_name, show_label=True)
387
-
388
- refiner_switch = gr.Slider(label='Refiner Switch At', minimum=0.1, maximum=1.0, step=0.0001,
389
- info='Use 0.4 for SD1.5 realistic models; '
390
- 'or 0.667 for SD1.5 anime models; '
391
- 'or 0.8 for XL-refiners; '
392
- 'or any value for switching two SDXL models.',
393
- value=modules.config.default_refiner_switch,
394
- visible=modules.config.default_refiner_model_name != 'None')
395
-
396
- refiner_model.change(lambda x: gr.update(visible=x != 'None'),
397
- inputs=refiner_model, outputs=refiner_switch, show_progress=False, queue=False)
398
-
399
- with gr.Group():
400
- lora_ctrls = []
401
-
402
- for i, (enabled, filename, weight) in enumerate(modules.config.default_loras):
403
- with gr.Row():
404
- lora_enabled = gr.Checkbox(label='Enable', value=enabled,
405
- elem_classes=['lora_enable', 'min_check'], scale=1)
406
- lora_model = gr.Dropdown(label=f'LoRA {i + 1}',
407
- choices=['None'] + modules.config.lora_filenames, value=filename,
408
- elem_classes='lora_model', scale=5)
409
- lora_weight = gr.Slider(label='Weight', minimum=modules.config.default_loras_min_weight,
410
- maximum=modules.config.default_loras_max_weight, step=0.01, value=weight,
411
- elem_classes='lora_weight', scale=5)
412
- lora_ctrls += [lora_enabled, lora_model, lora_weight]
413
 
414
- with gr.Row():
415
- refresh_files = gr.Button(label='Refresh', value='\U0001f504 Refresh All Files', variant='secondary', elem_classes='refresh_button')
416
- with gr.Tab(label='Advanced'):
417
- guidance_scale = gr.Slider(label='Guidance Scale', minimum=1.0, maximum=30.0, step=0.01,
418
- value=modules.config.default_cfg_scale,
419
- info='Higher value means style is cleaner, vivider, and more artistic.')
420
- sharpness = gr.Slider(label='Image Sharpness', minimum=0.0, maximum=30.0, step=0.001,
421
- value=modules.config.default_sample_sharpness,
422
- info='Higher value means image and texture are sharper.')
423
- gr.HTML('<a href="https://github.com/lllyasviel/Fooocus/discussions/117" target="_blank">\U0001F4D4 Document</a>')
424
- dev_mode = gr.Checkbox(label='Developer Debug Mode', value=False, container=False)
425
-
426
- with gr.Column(visible=False) as dev_tools:
427
- with gr.Tab(label='Debug Tools'):
428
- adm_scaler_positive = gr.Slider(label='Positive ADM Guidance Scaler', minimum=0.1, maximum=3.0,
429
- step=0.001, value=1.5, info='The scaler multiplied to positive ADM (use 1.0 to disable). ')
430
- adm_scaler_negative = gr.Slider(label='Negative ADM Guidance Scaler', minimum=0.1, maximum=3.0,
431
- step=0.001, value=0.8, info='The scaler multiplied to negative ADM (use 1.0 to disable). ')
432
- adm_scaler_end = gr.Slider(label='ADM Guidance End At Step', minimum=0.0, maximum=1.0,
433
- step=0.001, value=0.3,
434
- info='When to end the guidance from positive/negative ADM. ')
435
-
436
- refiner_swap_method = gr.Dropdown(label='Refiner swap method', value=flags.refiner_swap_method,
437
- choices=['joint', 'separate', 'vae'])
438
-
439
- adaptive_cfg = gr.Slider(label='CFG Mimicking from TSNR', minimum=1.0, maximum=30.0, step=0.01,
440
- value=modules.config.default_cfg_tsnr,
441
- info='Enabling Fooocus\'s implementation of CFG mimicking for TSNR '
442
- '(effective when real CFG > mimicked CFG).')
443
- sampler_name = gr.Dropdown(label='Sampler', choices=flags.sampler_list,
444
- value=modules.config.default_sampler)
445
- scheduler_name = gr.Dropdown(label='Scheduler', choices=flags.scheduler_list,
446
- value=modules.config.default_scheduler)
447
-
448
- generate_image_grid = gr.Checkbox(label='Generate Image Grid for Each Batch',
449
- info='(Experimental) This may cause performance problems on some computers and certain internet conditions.',
450
- value=False)
451
-
452
- overwrite_step = gr.Slider(label='Forced Overwrite of Sampling Step',
453
- minimum=-1, maximum=200, step=1,
454
- value=modules.config.default_overwrite_step,
455
- info='Set as -1 to disable. For developer debugging.')
456
- overwrite_switch = gr.Slider(label='Forced Overwrite of Refiner Switch Step',
457
- minimum=-1, maximum=200, step=1,
458
- value=modules.config.default_overwrite_switch,
459
- info='Set as -1 to disable. For developer debugging.')
460
- overwrite_width = gr.Slider(label='Forced Overwrite of Generating Width',
461
- minimum=-1, maximum=2048, step=1, value=-1,
462
- info='Set as -1 to disable. For developer debugging. '
463
- 'Results will be worse for non-standard numbers that SDXL is not trained on.')
464
- overwrite_height = gr.Slider(label='Forced Overwrite of Generating Height',
465
- minimum=-1, maximum=2048, step=1, value=-1,
466
- info='Set as -1 to disable. For developer debugging. '
467
- 'Results will be worse for non-standard numbers that SDXL is not trained on.')
468
- overwrite_vary_strength = gr.Slider(label='Forced Overwrite of Denoising Strength of "Vary"',
469
- minimum=-1, maximum=1.0, step=0.001, value=-1,
470
- info='Set as negative number to disable. For developer debugging.')
471
- overwrite_upscale_strength = gr.Slider(label='Forced Overwrite of Denoising Strength of "Upscale"',
472
- minimum=-1, maximum=1.0, step=0.001, value=-1,
473
- info='Set as negative number to disable. For developer debugging.')
474
- disable_preview = gr.Checkbox(label='Disable Preview', value=False,
475
- info='Disable preview during generation.')
476
- disable_intermediate_results = gr.Checkbox(label='Disable Intermediate Results',
477
- value=modules.config.default_performance == flags.Performance.EXTREME_SPEED.value,
478
- interactive=modules.config.default_performance != flags.Performance.EXTREME_SPEED.value,
479
- info='Disable intermediate results during generation, only show final gallery.')
480
- disable_seed_increment = gr.Checkbox(label='Disable seed increment',
481
- info='Disable automatic seed increment when image number is > 1.',
482
- value=False)
483
- read_wildcards_in_order = gr.Checkbox(label="Read wildcards in order", value=False)
484
-
485
- if not args_manager.args.disable_metadata:
486
- save_metadata_to_images = gr.Checkbox(label='Save Metadata to Images', value=modules.config.default_save_metadata_to_images,
487
- info='Adds parameters to generated images allowing manual regeneration.')
488
- metadata_scheme = gr.Radio(label='Metadata Scheme', choices=flags.metadata_scheme, value=modules.config.default_metadata_scheme,
489
- info='Image Prompt parameters are not included. Use png and a1111 for compatibility with Civitai.',
490
- visible=modules.config.default_save_metadata_to_images)
491
-
492
- save_metadata_to_images.change(lambda x: gr.update(visible=x), inputs=[save_metadata_to_images], outputs=[metadata_scheme],
493
- queue=False, show_progress=False)
494
-
495
- with gr.Tab(label='Control'):
496
- debugging_cn_preprocessor = gr.Checkbox(label='Debug Preprocessors', value=False,
497
- info='See the results from preprocessors.')
498
- skipping_cn_preprocessor = gr.Checkbox(label='Skip Preprocessors', value=False,
499
- info='Do not preprocess images. (Inputs are already canny/depth/cropped-face/etc.)')
500
-
501
- mixing_image_prompt_and_vary_upscale = gr.Checkbox(label='Mixing Image Prompt and Vary/Upscale',
502
- value=False)
503
- mixing_image_prompt_and_inpaint = gr.Checkbox(label='Mixing Image Prompt and Inpaint',
504
- value=False)
505
-
506
- controlnet_softness = gr.Slider(label='Softness of ControlNet', minimum=0.0, maximum=1.0,
507
- step=0.001, value=0.25,
508
- info='Similar to the Control Mode in A1111 (use 0.0 to disable). ')
509
-
510
- with gr.Tab(label='Canny'):
511
- canny_low_threshold = gr.Slider(label='Canny Low Threshold', minimum=1, maximum=255,
512
- step=1, value=64)
513
- canny_high_threshold = gr.Slider(label='Canny High Threshold', minimum=1, maximum=255,
514
- step=1, value=128)
515
-
516
- with gr.Tab(label='Inpaint'):
517
- debugging_inpaint_preprocessor = gr.Checkbox(label='Debug Inpaint Preprocessing', value=False)
518
- inpaint_disable_initial_latent = gr.Checkbox(label='Disable initial latent in inpaint', value=False)
519
- inpaint_engine = gr.Dropdown(label='Inpaint Engine',
520
- value=modules.config.default_inpaint_engine_version,
521
- choices=flags.inpaint_engine_versions,
522
- info='Version of Fooocus inpaint model')
523
- inpaint_strength = gr.Slider(label='Inpaint Denoising Strength',
524
- minimum=0.0, maximum=1.0, step=0.001, value=1.0,
525
- info='Same as the denoising strength in A1111 inpaint. '
526
- 'Only used in inpaint, not used in outpaint. '
527
- '(Outpaint always use 1.0)')
528
- inpaint_respective_field = gr.Slider(label='Inpaint Respective Field',
529
- minimum=0.0, maximum=1.0, step=0.001, value=0.618,
530
- info='The area to inpaint. '
531
- 'Value 0 is same as "Only Masked" in A1111. '
532
- 'Value 1 is same as "Whole Image" in A1111. '
533
- 'Only used in inpaint, not used in outpaint. '
534
- '(Outpaint always use 1.0)')
535
- inpaint_erode_or_dilate = gr.Slider(label='Mask Erode or Dilate',
536
- minimum=-64, maximum=64, step=1, value=0,
537
- info='Positive value will make white area in the mask larger, '
538
- 'negative value will make white area smaller.'
539
- '(default is 0, always process before any mask invert)')
540
- inpaint_mask_upload_checkbox = gr.Checkbox(label='Enable Mask Upload', value=False)
541
- invert_mask_checkbox = gr.Checkbox(label='Invert Mask', value=False)
542
-
543
- inpaint_ctrls = [debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine,
544
- inpaint_strength, inpaint_respective_field,
545
- inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate]
546
-
547
- inpaint_mask_upload_checkbox.change(lambda x: gr.update(visible=x),
548
- inputs=inpaint_mask_upload_checkbox,
549
- outputs=inpaint_mask_image, queue=False, show_progress=False)
550
-
551
- with gr.Tab(label='FreeU'):
552
- freeu_enabled = gr.Checkbox(label='Enabled', value=False)
553
- freeu_b1 = gr.Slider(label='B1', minimum=0, maximum=2, step=0.01, value=1.01)
554
- freeu_b2 = gr.Slider(label='B2', minimum=0, maximum=2, step=0.01, value=1.02)
555
- freeu_s1 = gr.Slider(label='S1', minimum=0, maximum=4, step=0.01, value=0.99)
556
- freeu_s2 = gr.Slider(label='S2', minimum=0, maximum=4, step=0.01, value=0.95)
557
- freeu_ctrls = [freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2]
558
-
559
- def dev_mode_checked(r):
560
- return gr.update(visible=r)
561
-
562
- dev_mode.change(dev_mode_checked, inputs=[dev_mode], outputs=[dev_tools],
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
563
  queue=False, show_progress=False)
564
 
565
- def refresh_files_clicked():
566
- modules.config.update_files()
567
- results = [gr.update(choices=modules.config.model_filenames)]
568
- results += [gr.update(choices=['None'] + modules.config.model_filenames)]
569
- if not args_manager.args.disable_preset_selection:
570
- results += [gr.update(choices=modules.config.available_presets)]
571
- for i in range(modules.config.default_max_lora_number):
572
- results += [gr.update(interactive=True),
573
- gr.update(choices=['None'] + modules.config.lora_filenames), gr.update()]
574
- return results
575
-
576
- refresh_files_output = [base_model, refiner_model]
577
- if not args_manager.args.disable_preset_selection:
578
- refresh_files_output += [preset_selection]
579
- refresh_files.click(refresh_files_clicked, [], refresh_files_output + lora_ctrls,
580
- queue=False, show_progress=False)
581
-
582
- state_is_generating = gr.State(False)
583
-
584
- load_data_outputs = [advanced_checkbox, image_number, prompt, negative_prompt, style_selections,
585
- performance_selection, overwrite_step, overwrite_switch, aspect_ratios_selection,
586
- overwrite_width, overwrite_height, guidance_scale, sharpness, adm_scaler_positive,
587
- adm_scaler_negative, adm_scaler_end, refiner_swap_method, adaptive_cfg, base_model,
588
- refiner_model, refiner_switch, sampler_name, scheduler_name, seed_random, image_seed,
589
- generate_button, load_parameter_button] + freeu_ctrls + lora_ctrls
590
-
591
- if not args_manager.args.disable_preset_selection:
592
- def preset_selection_change(preset, is_generating):
593
- preset_content = modules.config.try_get_preset_content(preset) if preset != 'initial' else {}
594
- preset_prepared = modules.meta_parser.parse_meta_from_preset(preset_content)
595
-
596
- default_model = preset_prepared.get('base_model')
597
- previous_default_models = preset_prepared.get('previous_default_models', [])
598
- checkpoint_downloads = preset_prepared.get('checkpoint_downloads', {})
599
- embeddings_downloads = preset_prepared.get('embeddings_downloads', {})
600
- lora_downloads = preset_prepared.get('lora_downloads', {})
601
-
602
- preset_prepared['base_model'], preset_prepared['lora_downloads'] = launch.download_models(
603
- default_model, previous_default_models, checkpoint_downloads, embeddings_downloads, lora_downloads)
604
-
605
- if 'prompt' in preset_prepared and preset_prepared.get('prompt') == '':
606
- del preset_prepared['prompt']
607
-
608
- return modules.meta_parser.load_parameter_button_click(json.dumps(preset_prepared), is_generating)
609
-
610
- preset_selection.change(preset_selection_change, inputs=[preset_selection, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=True) \
611
- .then(fn=style_sorter.sort_styles, inputs=style_selections, outputs=style_selections, queue=False, show_progress=False) \
612
-
613
- performance_selection.change(lambda x: [gr.update(interactive=not flags.Performance.has_restricted_features(x))] * 11 +
614
- [gr.update(visible=not flags.Performance.has_restricted_features(x))] * 1 +
615
- [gr.update(interactive=not flags.Performance.has_restricted_features(x), value=flags.Performance.has_restricted_features(x))] * 1,
616
- inputs=performance_selection,
617
- outputs=[
618
- guidance_scale, sharpness, adm_scaler_end, adm_scaler_positive,
619
- adm_scaler_negative, refiner_switch, refiner_model, sampler_name,
620
- scheduler_name, adaptive_cfg, refiner_swap_method, negative_prompt, disable_intermediate_results
621
- ], queue=False, show_progress=False)
622
-
623
- output_format.input(lambda x: gr.update(output_format=x), inputs=output_format)
624
-
625
- advanced_checkbox.change(lambda x: gr.update(visible=x), advanced_checkbox, advanced_column,
626
- queue=False, show_progress=False) \
627
- .then(fn=lambda: None, _js='refresh_grid_delayed', queue=False, show_progress=False)
628
-
629
- def inpaint_mode_change(mode):
630
- assert mode in modules.flags.inpaint_options
631
-
632
- # inpaint_additional_prompt, outpaint_selections, example_inpaint_prompts,
633
- # inpaint_disable_initial_latent, inpaint_engine,
634
- # inpaint_strength, inpaint_respective_field
635
-
636
- if mode == modules.flags.inpaint_option_detail:
637
- return [
638
- gr.update(visible=True), gr.update(visible=False, value=[]),
639
- gr.Dataset.update(visible=True, samples=modules.config.example_inpaint_prompts),
640
- False, 'None', 0.5, 0.0
641
- ]
642
-
643
- if mode == modules.flags.inpaint_option_modify:
644
- return [
645
- gr.update(visible=True), gr.update(visible=False, value=[]),
646
- gr.Dataset.update(visible=False, samples=modules.config.example_inpaint_prompts),
647
- True, modules.config.default_inpaint_engine_version, 1.0, 0.0
648
- ]
649
 
 
650
  return [
651
- gr.update(visible=False, value=''), gr.update(visible=True),
652
  gr.Dataset.update(visible=False, samples=modules.config.example_inpaint_prompts),
653
- False, modules.config.default_inpaint_engine_version, 1.0, 0.618
654
  ]
655
 
656
- inpaint_mode.input(inpaint_mode_change, inputs=inpaint_mode, outputs=[
657
- inpaint_additional_prompt, outpaint_selections, example_inpaint_prompts,
658
- inpaint_disable_initial_latent, inpaint_engine,
659
- inpaint_strength, inpaint_respective_field
660
- ], show_progress=False, queue=False)
661
-
662
- ctrls = [currentTask, generate_image_grid]
663
- ctrls += [
664
- prompt, negative_prompt, style_selections,
665
- performance_selection, aspect_ratios_selection, image_number, output_format, image_seed,
666
- read_wildcards_in_order, sharpness, guidance_scale
667
  ]
668
 
669
- ctrls += [base_model, refiner_model, refiner_switch] + lora_ctrls
670
- ctrls += [input_image_checkbox, current_tab]
671
- ctrls += [uov_method, uov_input_image]
672
- ctrls += [outpaint_selections, inpaint_input_image, inpaint_additional_prompt, inpaint_mask_image]
673
- ctrls += [disable_preview, disable_intermediate_results, disable_seed_increment]
674
- ctrls += [adm_scaler_positive, adm_scaler_negative, adm_scaler_end, adaptive_cfg]
675
- ctrls += [sampler_name, scheduler_name]
676
- ctrls += [overwrite_step, overwrite_switch, overwrite_width, overwrite_height, overwrite_vary_strength]
677
- ctrls += [overwrite_upscale_strength, mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint]
678
- ctrls += [debugging_cn_preprocessor, skipping_cn_preprocessor, canny_low_threshold, canny_high_threshold]
679
- ctrls += [refiner_swap_method, controlnet_softness]
680
- ctrls += freeu_ctrls
681
- ctrls += inpaint_ctrls
682
-
683
- if not args_manager.args.disable_metadata:
684
- ctrls += [save_metadata_to_images, metadata_scheme]
685
-
686
- ctrls += ip_ctrls
687
-
688
- def parse_meta(raw_prompt_txt, is_generating):
689
- loaded_json = None
690
- if is_json(raw_prompt_txt):
691
- loaded_json = json.loads(raw_prompt_txt)
692
-
693
- if loaded_json is None:
694
- if is_generating:
695
- return gr.update(), gr.update(), gr.update()
696
- else:
697
- return gr.update(), gr.update(visible=True), gr.update(visible=False)
698
-
699
- return json.dumps(loaded_json), gr.update(visible=False), gr.update(visible=True)
700
-
701
- prompt.input(parse_meta, inputs=[prompt, state_is_generating], outputs=[prompt, generate_button, load_parameter_button], queue=False, show_progress=False)
702
-
703
- load_parameter_button.click(modules.meta_parser.load_parameter_button_click, inputs=[prompt, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=False)
704
-
705
- def trigger_metadata_import(filepath, state_is_generating):
706
- parameters, metadata_scheme = modules.meta_parser.read_info_from_image(filepath)
707
- if parameters is None:
708
- print('Could not find metadata in the image!')
709
- parsed_parameters = {}
710
  else:
711
- metadata_parser = modules.meta_parser.get_metadata_parser(metadata_scheme)
712
- parsed_parameters = metadata_parser.parse_json(parameters)
713
-
714
- return modules.meta_parser.load_parameter_button_click(parsed_parameters, state_is_generating)
715
-
716
- metadata_import_button.click(trigger_metadata_import, inputs=[metadata_input_image, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=True) \
717
- .then(style_sorter.sort_styles, inputs=style_selections, outputs=style_selections, queue=False, show_progress=False)
718
-
719
- generate_button.click(lambda: (gr.update(visible=True, interactive=True), gr.update(visible=True, interactive=True), gr.update(visible=False, interactive=False), [], True),
720
- outputs=[stop_button, skip_button, generate_button, gallery, state_is_generating]) \
721
- .then(fn=refresh_seed, inputs=[seed_random, image_seed], outputs=image_seed) \
722
- .then(fn=get_task, inputs=ctrls, outputs=currentTask) \
723
- .then(fn=generate_clicked, inputs=currentTask, outputs=[progress_html, progress_window, progress_gallery, gallery]) \
724
- .then(lambda: (gr.update(visible=True, interactive=True), gr.update(visible=False, interactive=False), gr.update(visible=False, interactive=False), False),
725
- outputs=[generate_button, stop_button, skip_button, state_is_generating]) \
726
- .then(fn=update_history_link, outputs=history_link) \
727
- .then(fn=lambda: None, _js='playNotification').then(fn=lambda: None, _js='refresh_grid_delayed')
728
-
729
- for notification_file in ['notification.ogg', 'notification.mp3']:
730
- if os.path.exists(notification_file):
731
- gr.Audio(interactive=False, value=notification_file, elem_id='audio_notification', visible=False)
732
- break
733
-
734
- def trigger_describe(mode, img):
735
- if mode == flags.desc_type_photo:
736
- from extras.interrogate import default_interrogator as default_interrogator_photo
737
- return default_interrogator_photo(img), ["Fooocus V2", "Fooocus Enhance", "Fooocus Sharp"]
738
- if mode == flags.desc_type_anime:
739
- from extras.wd14tagger import default_interrogator as default_interrogator_anime
740
- return default_interrogator_anime(img), ["Fooocus V2", "Fooocus Masterpiece"]
741
- return mode, ["Fooocus V2"]
742
-
743
- desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image],
744
- outputs=[prompt, style_selections], show_progress=True, queue=True)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
745
 
746
 
747
  def dump_default_english_config():
 
16
  import args_manager
17
  import copy
18
  import launch
 
19
 
20
  from modules.sdxl_styles import legal_style_names
21
  from modules.private_logger import get_current_html_path
 
23
  from modules.auth import auth_enabled, check_auth
24
  from modules.util import is_json
25
 
 
 
 
 
 
 
 
 
 
26
  def get_task(*args):
27
  args = list(args)
28
  args.pop(0)
 
96
  if isinstance(args_manager.args.preset, str):
97
  title += ' ' + args_manager.args.preset
98
 
99
+ shared.gradio_root = gr.Blocks(title=title).queue()
 
 
 
 
 
 
 
100
 
101
  with shared.gradio_root:
102
+ currentTask = gr.State(worker.AsyncTask(args=[]))
103
+ with gr.Row():
104
+ with gr.Column(scale=2):
105
+ with gr.Row():
106
+ progress_window = grh.Image(label='Preview', show_label=True, visible=False, height=768,
107
+ elem_classes=['main_view'])
108
+ progress_gallery = gr.Gallery(label='Finished Images', show_label=True, object_fit='contain',
109
+ height=768, visible=False, elem_classes=['main_view', 'image_gallery'])
110
+ progress_html = gr.HTML(value=modules.html.make_progress_html(32, 'Progress 32%'), visible=False,
111
+ elem_id='progress-bar', elem_classes='progress-bar')
112
+ gallery = gr.Gallery(label='Gallery', show_label=False, object_fit='contain', visible=True, height=768,
113
+ elem_classes=['resizable_area', 'main_view', 'final_gallery', 'image_gallery'],
114
+ elem_id='final_gallery')
115
+ with gr.Row(elem_classes='type_row'):
116
+ with gr.Column(scale=17):
117
+ prompt = gr.Textbox(show_label=False, placeholder="Type prompt here or paste parameters.", elem_id='positive_prompt',
118
+ container=False, autofocus=True, elem_classes='type_row', lines=1024)
119
+
120
+ default_prompt = modules.config.default_prompt
121
+ if isinstance(default_prompt, str) and default_prompt != '':
122
+ shared.gradio_root.load(lambda: default_prompt, outputs=prompt)
123
+
124
+ with gr.Column(scale=3, min_width=0):
125
+ generate_button = gr.Button(label="Generate", value="Generate", elem_classes='type_row', elem_id='generate_button', visible=True)
126
+ load_parameter_button = gr.Button(label="Load Parameters", value="Load Parameters", elem_classes='type_row', elem_id='load_parameter_button', visible=False)
127
+ skip_button = gr.Button(label="Skip", value="Skip", elem_classes='type_row_half', visible=False)
128
+ stop_button = gr.Button(label="Stop", value="Stop", elem_classes='type_row_half', elem_id='stop_button', visible=False)
129
+
130
+ def stop_clicked(currentTask):
131
+ import ldm_patched.modules.model_management as model_management
132
+ currentTask.last_stop = 'stop'
133
+ if (currentTask.processing):
134
+ model_management.interrupt_current_processing()
135
+ return currentTask
136
+
137
+ def skip_clicked(currentTask):
138
+ import ldm_patched.modules.model_management as model_management
139
+ currentTask.last_stop = 'skip'
140
+ if (currentTask.processing):
141
+ model_management.interrupt_current_processing()
142
+ return currentTask
143
+
144
+ stop_button.click(stop_clicked, inputs=currentTask, outputs=currentTask, queue=False, show_progress=False, _js='cancelGenerateForever')
145
+ skip_button.click(skip_clicked, inputs=currentTask, outputs=currentTask, queue=False, show_progress=False)
146
+ with gr.Row(elem_classes='advanced_check_row'):
147
+ input_image_checkbox = gr.Checkbox(label='Input Image', value=False, container=False, elem_classes='min_check')
148
+ advanced_checkbox = gr.Checkbox(label='Advanced', value=modules.config.default_advanced_checkbox, container=False, elem_classes='min_check')
149
+ with gr.Row(visible=False) as image_input_panel:
150
+ with gr.Tabs():
151
+ with gr.TabItem(label='Upscale or Variation') as uov_tab:
152
+ with gr.Row():
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
153
  with gr.Column():
154
+ uov_input_image = grh.Image(label='Drag above image to here', source='upload', type='numpy')
155
+ with gr.Column():
156
+ uov_method = gr.Radio(label='Upscale or Variation:', choices=flags.uov_list, value=flags.disabled)
157
+ gr.HTML('<a href="https://github.com/lllyasviel/Fooocus/discussions/390" target="_blank">\U0001F4D4 Document</a>')
158
+ with gr.TabItem(label='Image Prompt') as ip_tab:
159
+ with gr.Row():
160
+ ip_images = []
161
+ ip_types = []
162
+ ip_stops = []
163
+ ip_weights = []
164
+ ip_ctrls = []
165
+ ip_ad_cols = []
166
+ for _ in range(flags.controlnet_image_count):
167
+ with gr.Column():
168
+ ip_image = grh.Image(label='Image', source='upload', type='numpy', show_label=False, height=300)
169
+ ip_images.append(ip_image)
170
+ ip_ctrls.append(ip_image)
171
+ with gr.Column(visible=False) as ad_col:
172
+ with gr.Row():
173
+ default_end, default_weight = flags.default_parameters[flags.default_ip]
174
+
175
+ ip_stop = gr.Slider(label='Stop At', minimum=0.0, maximum=1.0, step=0.001, value=default_end)
176
+ ip_stops.append(ip_stop)
177
+ ip_ctrls.append(ip_stop)
178
+
179
+ ip_weight = gr.Slider(label='Weight', minimum=0.0, maximum=2.0, step=0.001, value=default_weight)
180
+ ip_weights.append(ip_weight)
181
+ ip_ctrls.append(ip_weight)
182
+
183
+ ip_type = gr.Radio(label='Type', choices=flags.ip_list, value=flags.default_ip, container=False)
184
+ ip_types.append(ip_type)
185
+ ip_ctrls.append(ip_type)
186
+
187
+ ip_type.change(lambda x: flags.default_parameters[x], inputs=[ip_type], outputs=[ip_stop, ip_weight], queue=False, show_progress=False)
188
+ ip_ad_cols.append(ad_col)
189
+ ip_advanced = gr.Checkbox(label='Advanced', value=False, container=False)
190
+ gr.HTML('* \"Image Prompt\" is powered by Fooocus Image Mixture Engine (v1.0.1). <a href="https://github.com/lllyasviel/Fooocus/discussions/557" target="_blank">\U0001F4D4 Document</a>')
191
+
192
+ def ip_advance_checked(x):
193
+ return [gr.update(visible=x)] * len(ip_ad_cols) + \
194
+ [flags.default_ip] * len(ip_types) + \
195
+ [flags.default_parameters[flags.default_ip][0]] * len(ip_stops) + \
196
+ [flags.default_parameters[flags.default_ip][1]] * len(ip_weights)
197
+
198
+ ip_advanced.change(ip_advance_checked, inputs=ip_advanced,
199
+ outputs=ip_ad_cols + ip_types + ip_stops + ip_weights,
200
+ queue=False, show_progress=False)
201
+ with gr.TabItem(label='Inpaint or Outpaint') as inpaint_tab:
202
+ with gr.Row():
203
+ inpaint_input_image = grh.Image(label='Drag inpaint or outpaint image to here', source='upload', type='numpy', tool='sketch', height=500, brush_color="#FFFFFF", elem_id='inpaint_canvas')
204
+ inpaint_mask_image = grh.Image(label='Mask Upload', source='upload', type='numpy', height=500, visible=False)
205
 
206
+ with gr.Row():
207
+ inpaint_additional_prompt = gr.Textbox(placeholder="Describe what you want to inpaint.", elem_id='inpaint_additional_prompt', label='Inpaint Additional Prompt', visible=False)
208
+ outpaint_selections = gr.CheckboxGroup(choices=['Left', 'Right', 'Top', 'Bottom'], value=[], label='Outpaint Direction')
209
+ inpaint_mode = gr.Dropdown(choices=modules.flags.inpaint_options, value=modules.flags.inpaint_option_default, label='Method')
210
+ example_inpaint_prompts = gr.Dataset(samples=modules.config.example_inpaint_prompts, label='Additional Prompt Quick List', components=[inpaint_additional_prompt], visible=False)
211
+ gr.HTML('* Powered by Fooocus Inpaint Engine <a href="https://github.com/lllyasviel/Fooocus/discussions/414" target="_blank">\U0001F4D4 Document</a>')
212
+ example_inpaint_prompts.click(lambda x: x[0], inputs=example_inpaint_prompts, outputs=inpaint_additional_prompt, show_progress=False, queue=False)
213
+ with gr.TabItem(label='Describe') as desc_tab:
214
+ with gr.Row():
215
+ with gr.Column():
216
+ desc_input_image = grh.Image(label='Drag any image to here', source='upload', type='numpy')
217
+ with gr.Column():
218
+ desc_method = gr.Radio(
219
+ label='Content Type',
220
+ choices=[flags.desc_type_photo, flags.desc_type_anime],
221
+ value=flags.desc_type_photo)
222
+ desc_btn = gr.Button(value='Describe this Image into Prompt')
223
+ gr.HTML('<a href="https://github.com/lllyasviel/Fooocus/discussions/1363" target="_blank">\U0001F4D4 Document</a>')
224
+ with gr.TabItem(label='Metadata') as load_tab:
225
+ with gr.Column():
226
+ metadata_input_image = grh.Image(label='Drag any image generated by Fooocus here', source='upload', type='filepath')
227
+ metadata_json = gr.JSON(label='Metadata')
228
+ metadata_import_button = gr.Button(value='Apply Metadata')
229
+
230
+ def trigger_metadata_preview(filepath):
231
+ parameters, metadata_scheme = modules.meta_parser.read_info_from_image(filepath)
232
+
233
+ results = {}
234
+ if parameters is not None:
235
+ results['parameters'] = parameters
236
+
237
+ if isinstance(metadata_scheme, flags.MetadataScheme):
238
+ results['metadata_scheme'] = metadata_scheme.value
239
+
240
+ return results
241
+
242
+ metadata_input_image.upload(trigger_metadata_preview, inputs=metadata_input_image,
243
+ outputs=metadata_json, queue=False, show_progress=True)
244
+
245
+ switch_js = "(x) => {if(x){viewer_to_bottom(100);viewer_to_bottom(500);}else{viewer_to_top();} return x;}"
246
+ down_js = "() => {viewer_to_bottom();}"
247
+
248
+ input_image_checkbox.change(lambda x: gr.update(visible=x), inputs=input_image_checkbox,
249
+ outputs=image_input_panel, queue=False, show_progress=False, _js=switch_js)
250
+ ip_advanced.change(lambda: None, queue=False, show_progress=False, _js=down_js)
251
+
252
+ current_tab = gr.Textbox(value='uov', visible=False)
253
+ uov_tab.select(lambda: 'uov', outputs=current_tab, queue=False, _js=down_js, show_progress=False)
254
+ inpaint_tab.select(lambda: 'inpaint', outputs=current_tab, queue=False, _js=down_js, show_progress=False)
255
+ ip_tab.select(lambda: 'ip', outputs=current_tab, queue=False, _js=down_js, show_progress=False)
256
+ desc_tab.select(lambda: 'desc', outputs=current_tab, queue=False, _js=down_js, show_progress=False)
257
+
258
+ with gr.Column(scale=1, visible=modules.config.default_advanced_checkbox) as advanced_column:
259
+ with gr.Tab(label='Setting'):
260
+ if not args_manager.args.disable_preset_selection:
261
+ preset_selection = gr.Radio(label='Preset',
262
+ choices=modules.config.available_presets,
263
+ value=args_manager.args.preset if args_manager.args.preset else "initial",
264
+ interactive=True)
265
+ performance_selection = gr.Radio(label='Performance',
266
+ choices=flags.Performance.list(),
267
+ value=modules.config.default_performance)
268
+ aspect_ratios_selection = gr.Radio(label='Aspect Ratios', choices=modules.config.available_aspect_ratios,
269
+ value=modules.config.default_aspect_ratio, info='width × height',
270
+ elem_classes='aspect_ratios')
271
+ image_number = gr.Slider(label='Image Number', minimum=1, maximum=modules.config.default_max_image_number, step=1, value=modules.config.default_image_number)
272
+
273
+ output_format = gr.Radio(label='Output Format',
274
+ choices=flags.OutputFormat.list(),
275
+ value=modules.config.default_output_format)
276
+
277
+ negative_prompt = gr.Textbox(label='Negative Prompt', show_label=True, placeholder="Type prompt here.",
278
+ info='Describing what you do not want to see.', lines=2,
279
+ elem_id='negative_prompt',
280
+ value=modules.config.default_prompt_negative)
281
+ seed_random = gr.Checkbox(label='Random', value=True)
282
+ image_seed = gr.Textbox(label='Seed', value=0, max_lines=1, visible=False) # workaround for https://github.com/gradio-app/gradio/issues/5354
283
+
284
+ def random_checked(r):
285
+ return gr.update(visible=not r)
286
+
287
+ def refresh_seed(r, seed_string):
288
+ if r:
289
+ return random.randint(constants.MIN_SEED, constants.MAX_SEED)
290
+ else:
291
+ try:
292
+ seed_value = int(seed_string)
293
+ if constants.MIN_SEED <= seed_value <= constants.MAX_SEED:
294
+ return seed_value
295
+ except ValueError:
296
+ pass
297
+ return random.randint(constants.MIN_SEED, constants.MAX_SEED)
298
+
299
+ seed_random.change(random_checked, inputs=[seed_random], outputs=[image_seed],
300
+ queue=False, show_progress=False)
301
+
302
+ def update_history_link():
303
+ if args_manager.args.disable_image_log:
304
+ return gr.update(value='')
305
+
306
+ return gr.update(value=f'<a href="file={get_current_html_path(output_format)}" target="_blank">\U0001F4DA History Log</a>')
307
+
308
+ history_link = gr.HTML()
309
+ shared.gradio_root.load(update_history_link, outputs=history_link, queue=False, show_progress=False)
310
+
311
+ with gr.Tab(label='Style', elem_classes=['style_selections_tab']):
312
+ style_sorter.try_load_sorted_styles(
313
+ style_names=legal_style_names,
314
+ default_selected=modules.config.default_styles)
315
+
316
+ style_search_bar = gr.Textbox(show_label=False, container=False,
317
+ placeholder="\U0001F50E Type here to search styles ...",
318
+ value="",
319
+ label='Search Styles')
320
+ style_selections = gr.CheckboxGroup(show_label=False, container=False,
321
+ choices=copy.deepcopy(style_sorter.all_styles),
322
+ value=copy.deepcopy(modules.config.default_styles),
323
+ label='Selected Styles',
324
+ elem_classes=['style_selections'])
325
+ gradio_receiver_style_selections = gr.Textbox(elem_id='gradio_receiver_style_selections', visible=False)
326
+
327
+ shared.gradio_root.load(lambda: gr.update(choices=copy.deepcopy(style_sorter.all_styles)),
328
+ outputs=style_selections)
329
+
330
+ style_search_bar.change(style_sorter.search_styles,
331
+ inputs=[style_selections, style_search_bar],
332
+ outputs=style_selections,
333
+ queue=False,
334
+ show_progress=False).then(
335
+ lambda: None, _js='()=>{refresh_style_localization();}')
336
+
337
+ gradio_receiver_style_selections.input(style_sorter.sort_styles,
338
+ inputs=style_selections,
339
+ outputs=style_selections,
340
+ queue=False,
341
+ show_progress=False).then(
342
+ lambda: None, _js='()=>{refresh_style_localization();}')
343
+
344
+ with gr.Tab(label='Model'):
345
+ with gr.Group():
346
+ with gr.Row():
347
+ base_model = gr.Dropdown(label='Base Model (SDXL only)', choices=modules.config.model_filenames, value=modules.config.default_base_model_name, show_label=True)
348
+ refiner_model = gr.Dropdown(label='Refiner (SDXL or SD 1.5)', choices=['None'] + modules.config.model_filenames, value=modules.config.default_refiner_model_name, show_label=True)
349
 
350
+ refiner_switch = gr.Slider(label='Refiner Switch At', minimum=0.1, maximum=1.0, step=0.0001,
351
+ info='Use 0.4 for SD1.5 realistic models; '
352
+ 'or 0.667 for SD1.5 anime models; '
353
+ 'or 0.8 for XL-refiners; '
354
+ 'or any value for switching two SDXL models.',
355
+ value=modules.config.default_refiner_switch,
356
+ visible=modules.config.default_refiner_model_name != 'None')
357
 
358
+ refiner_model.change(lambda x: gr.update(visible=x != 'None'),
359
+ inputs=refiner_model, outputs=refiner_switch, show_progress=False, queue=False)
 
 
 
360
 
361
+ with gr.Group():
362
+ lora_ctrls = []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
363
 
364
+ for i, (enabled, filename, weight) in enumerate(modules.config.default_loras):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
365
  with gr.Row():
366
+ lora_enabled = gr.Checkbox(label='Enable', value=enabled,
367
+ elem_classes=['lora_enable', 'min_check'], scale=1)
368
+ lora_model = gr.Dropdown(label=f'LoRA {i + 1}',
369
+ choices=['None'] + modules.config.lora_filenames, value=filename,
370
+ elem_classes='lora_model', scale=5)
371
+ lora_weight = gr.Slider(label='Weight', minimum=modules.config.default_loras_min_weight,
372
+ maximum=modules.config.default_loras_max_weight, step=0.01, value=weight,
373
+ elem_classes='lora_weight', scale=5)
374
+ lora_ctrls += [lora_enabled, lora_model, lora_weight]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
375
 
376
+ with gr.Row():
377
+ refresh_files = gr.Button(label='Refresh', value='\U0001f504 Refresh All Files', variant='secondary', elem_classes='refresh_button')
378
+ with gr.Tab(label='Advanced'):
379
+ guidance_scale = gr.Slider(label='Guidance Scale', minimum=1.0, maximum=30.0, step=0.01,
380
+ value=modules.config.default_cfg_scale,
381
+ info='Higher value means style is cleaner, vivider, and more artistic.')
382
+ sharpness = gr.Slider(label='Image Sharpness', minimum=0.0, maximum=30.0, step=0.001,
383
+ value=modules.config.default_sample_sharpness,
384
+ info='Higher value means image and texture are sharper.')
385
+ gr.HTML('<a href="https://github.com/lllyasviel/Fooocus/discussions/117" target="_blank">\U0001F4D4 Document</a>')
386
+ dev_mode = gr.Checkbox(label='Developer Debug Mode', value=False, container=False)
387
+
388
+ with gr.Column(visible=False) as dev_tools:
389
+ with gr.Tab(label='Debug Tools'):
390
+ adm_scaler_positive = gr.Slider(label='Positive ADM Guidance Scaler', minimum=0.1, maximum=3.0,
391
+ step=0.001, value=1.5, info='The scaler multiplied to positive ADM (use 1.0 to disable). ')
392
+ adm_scaler_negative = gr.Slider(label='Negative ADM Guidance Scaler', minimum=0.1, maximum=3.0,
393
+ step=0.001, value=0.8, info='The scaler multiplied to negative ADM (use 1.0 to disable). ')
394
+ adm_scaler_end = gr.Slider(label='ADM Guidance End At Step', minimum=0.0, maximum=1.0,
395
+ step=0.001, value=0.3,
396
+ info='When to end the guidance from positive/negative ADM. ')
397
+
398
+ refiner_swap_method = gr.Dropdown(label='Refiner swap method', value=flags.refiner_swap_method,
399
+ choices=['joint', 'separate', 'vae'])
400
+
401
+ adaptive_cfg = gr.Slider(label='CFG Mimicking from TSNR', minimum=1.0, maximum=30.0, step=0.01,
402
+ value=modules.config.default_cfg_tsnr,
403
+ info='Enabling Fooocus\'s implementation of CFG mimicking for TSNR '
404
+ '(effective when real CFG > mimicked CFG).')
405
+ sampler_name = gr.Dropdown(label='Sampler', choices=flags.sampler_list,
406
+ value=modules.config.default_sampler)
407
+ scheduler_name = gr.Dropdown(label='Scheduler', choices=flags.scheduler_list,
408
+ value=modules.config.default_scheduler)
409
+
410
+ generate_image_grid = gr.Checkbox(label='Generate Image Grid for Each Batch',
411
+ info='(Experimental) This may cause performance problems on some computers and certain internet conditions.',
412
+ value=False)
413
+
414
+ overwrite_step = gr.Slider(label='Forced Overwrite of Sampling Step',
415
+ minimum=-1, maximum=200, step=1,
416
+ value=modules.config.default_overwrite_step,
417
+ info='Set as -1 to disable. For developer debugging.')
418
+ overwrite_switch = gr.Slider(label='Forced Overwrite of Refiner Switch Step',
419
+ minimum=-1, maximum=200, step=1,
420
+ value=modules.config.default_overwrite_switch,
421
+ info='Set as -1 to disable. For developer debugging.')
422
+ overwrite_width = gr.Slider(label='Forced Overwrite of Generating Width',
423
+ minimum=-1, maximum=2048, step=1, value=-1,
424
+ info='Set as -1 to disable. For developer debugging. '
425
+ 'Results will be worse for non-standard numbers that SDXL is not trained on.')
426
+ overwrite_height = gr.Slider(label='Forced Overwrite of Generating Height',
427
+ minimum=-1, maximum=2048, step=1, value=-1,
428
+ info='Set as -1 to disable. For developer debugging. '
429
+ 'Results will be worse for non-standard numbers that SDXL is not trained on.')
430
+ overwrite_vary_strength = gr.Slider(label='Forced Overwrite of Denoising Strength of "Vary"',
431
+ minimum=-1, maximum=1.0, step=0.001, value=-1,
432
+ info='Set as negative number to disable. For developer debugging.')
433
+ overwrite_upscale_strength = gr.Slider(label='Forced Overwrite of Denoising Strength of "Upscale"',
434
+ minimum=-1, maximum=1.0, step=0.001, value=-1,
435
+ info='Set as negative number to disable. For developer debugging.')
436
+ disable_preview = gr.Checkbox(label='Disable Preview', value=False,
437
+ info='Disable preview during generation.')
438
+ disable_intermediate_results = gr.Checkbox(label='Disable Intermediate Results',
439
+ value=modules.config.default_performance == flags.Performance.EXTREME_SPEED.value,
440
+ interactive=modules.config.default_performance != flags.Performance.EXTREME_SPEED.value,
441
+ info='Disable intermediate results during generation, only show final gallery.')
442
+ disable_seed_increment = gr.Checkbox(label='Disable seed increment',
443
+ info='Disable automatic seed increment when image number is > 1.',
444
+ value=False)
445
+ read_wildcards_in_order = gr.Checkbox(label="Read wildcards in order", value=False)
446
+
447
+ if not args_manager.args.disable_metadata:
448
+ save_metadata_to_images = gr.Checkbox(label='Save Metadata to Images', value=modules.config.default_save_metadata_to_images,
449
+ info='Adds parameters to generated images allowing manual regeneration.')
450
+ metadata_scheme = gr.Radio(label='Metadata Scheme', choices=flags.metadata_scheme, value=modules.config.default_metadata_scheme,
451
+ info='Image Prompt parameters are not included. Use png and a1111 for compatibility with Civitai.',
452
+ visible=modules.config.default_save_metadata_to_images)
453
+
454
+ save_metadata_to_images.change(lambda x: gr.update(visible=x), inputs=[save_metadata_to_images], outputs=[metadata_scheme],
455
+ queue=False, show_progress=False)
456
+
457
+ with gr.Tab(label='Control'):
458
+ debugging_cn_preprocessor = gr.Checkbox(label='Debug Preprocessors', value=False,
459
+ info='See the results from preprocessors.')
460
+ skipping_cn_preprocessor = gr.Checkbox(label='Skip Preprocessors', value=False,
461
+ info='Do not preprocess images. (Inputs are already canny/depth/cropped-face/etc.)')
462
+
463
+ mixing_image_prompt_and_vary_upscale = gr.Checkbox(label='Mixing Image Prompt and Vary/Upscale',
464
+ value=False)
465
+ mixing_image_prompt_and_inpaint = gr.Checkbox(label='Mixing Image Prompt and Inpaint',
466
+ value=False)
467
+
468
+ controlnet_softness = gr.Slider(label='Softness of ControlNet', minimum=0.0, maximum=1.0,
469
+ step=0.001, value=0.25,
470
+ info='Similar to the Control Mode in A1111 (use 0.0 to disable). ')
471
+
472
+ with gr.Tab(label='Canny'):
473
+ canny_low_threshold = gr.Slider(label='Canny Low Threshold', minimum=1, maximum=255,
474
+ step=1, value=64)
475
+ canny_high_threshold = gr.Slider(label='Canny High Threshold', minimum=1, maximum=255,
476
+ step=1, value=128)
477
+
478
+ with gr.Tab(label='Inpaint'):
479
+ debugging_inpaint_preprocessor = gr.Checkbox(label='Debug Inpaint Preprocessing', value=False)
480
+ inpaint_disable_initial_latent = gr.Checkbox(label='Disable initial latent in inpaint', value=False)
481
+ inpaint_engine = gr.Dropdown(label='Inpaint Engine',
482
+ value=modules.config.default_inpaint_engine_version,
483
+ choices=flags.inpaint_engine_versions,
484
+ info='Version of Fooocus inpaint model')
485
+ inpaint_strength = gr.Slider(label='Inpaint Denoising Strength',
486
+ minimum=0.0, maximum=1.0, step=0.001, value=1.0,
487
+ info='Same as the denoising strength in A1111 inpaint. '
488
+ 'Only used in inpaint, not used in outpaint. '
489
+ '(Outpaint always use 1.0)')
490
+ inpaint_respective_field = gr.Slider(label='Inpaint Respective Field',
491
+ minimum=0.0, maximum=1.0, step=0.001, value=0.618,
492
+ info='The area to inpaint. '
493
+ 'Value 0 is same as "Only Masked" in A1111. '
494
+ 'Value 1 is same as "Whole Image" in A1111. '
495
+ 'Only used in inpaint, not used in outpaint. '
496
+ '(Outpaint always use 1.0)')
497
+ inpaint_erode_or_dilate = gr.Slider(label='Mask Erode or Dilate',
498
+ minimum=-64, maximum=64, step=1, value=0,
499
+ info='Positive value will make white area in the mask larger, '
500
+ 'negative value will make white area smaller.'
501
+ '(default is 0, always process before any mask invert)')
502
+ inpaint_mask_upload_checkbox = gr.Checkbox(label='Enable Mask Upload', value=False)
503
+ invert_mask_checkbox = gr.Checkbox(label='Invert Mask', value=False)
504
+
505
+ inpaint_ctrls = [debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine,
506
+ inpaint_strength, inpaint_respective_field,
507
+ inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate]
508
+
509
+ inpaint_mask_upload_checkbox.change(lambda x: gr.update(visible=x),
510
+ inputs=inpaint_mask_upload_checkbox,
511
+ outputs=inpaint_mask_image, queue=False, show_progress=False)
512
+
513
+ with gr.Tab(label='FreeU'):
514
+ freeu_enabled = gr.Checkbox(label='Enabled', value=False)
515
+ freeu_b1 = gr.Slider(label='B1', minimum=0, maximum=2, step=0.01, value=1.01)
516
+ freeu_b2 = gr.Slider(label='B2', minimum=0, maximum=2, step=0.01, value=1.02)
517
+ freeu_s1 = gr.Slider(label='S1', minimum=0, maximum=4, step=0.01, value=0.99)
518
+ freeu_s2 = gr.Slider(label='S2', minimum=0, maximum=4, step=0.01, value=0.95)
519
+ freeu_ctrls = [freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2]
520
+
521
+ def dev_mode_checked(r):
522
+ return gr.update(visible=r)
523
+
524
+ dev_mode.change(dev_mode_checked, inputs=[dev_mode], outputs=[dev_tools],
525
+ queue=False, show_progress=False)
526
+
527
+ def refresh_files_clicked():
528
+ modules.config.update_files()
529
+ results = [gr.update(choices=modules.config.model_filenames)]
530
+ results += [gr.update(choices=['None'] + modules.config.model_filenames)]
531
+ if not args_manager.args.disable_preset_selection:
532
+ results += [gr.update(choices=modules.config.available_presets)]
533
+ for i in range(modules.config.default_max_lora_number):
534
+ results += [gr.update(interactive=True),
535
+ gr.update(choices=['None'] + modules.config.lora_filenames), gr.update()]
536
+ return results
537
+
538
+ refresh_files_output = [base_model, refiner_model]
539
+ if not args_manager.args.disable_preset_selection:
540
+ refresh_files_output += [preset_selection]
541
+ refresh_files.click(refresh_files_clicked, [], refresh_files_output + lora_ctrls,
542
  queue=False, show_progress=False)
543
 
544
+ state_is_generating = gr.State(False)
545
+
546
+ load_data_outputs = [advanced_checkbox, image_number, prompt, negative_prompt, style_selections,
547
+ performance_selection, overwrite_step, overwrite_switch, aspect_ratios_selection,
548
+ overwrite_width, overwrite_height, guidance_scale, sharpness, adm_scaler_positive,
549
+ adm_scaler_negative, adm_scaler_end, refiner_swap_method, adaptive_cfg, base_model,
550
+ refiner_model, refiner_switch, sampler_name, scheduler_name, seed_random, image_seed,
551
+ generate_button, load_parameter_button] + freeu_ctrls + lora_ctrls
552
+
553
+ if not args_manager.args.disable_preset_selection:
554
+ def preset_selection_change(preset, is_generating):
555
+ preset_content = modules.config.try_get_preset_content(preset) if preset != 'initial' else {}
556
+ preset_prepared = modules.meta_parser.parse_meta_from_preset(preset_content)
557
+
558
+ default_model = preset_prepared.get('base_model')
559
+ previous_default_models = preset_prepared.get('previous_default_models', [])
560
+ checkpoint_downloads = preset_prepared.get('checkpoint_downloads', {})
561
+ embeddings_downloads = preset_prepared.get('embeddings_downloads', {})
562
+ lora_downloads = preset_prepared.get('lora_downloads', {})
563
+
564
+ preset_prepared['base_model'], preset_prepared['lora_downloads'] = launch.download_models(
565
+ default_model, previous_default_models, checkpoint_downloads, embeddings_downloads, lora_downloads)
566
+
567
+ if 'prompt' in preset_prepared and preset_prepared.get('prompt') == '':
568
+ del preset_prepared['prompt']
569
+
570
+ return modules.meta_parser.load_parameter_button_click(json.dumps(preset_prepared), is_generating)
571
+
572
+ preset_selection.change(preset_selection_change, inputs=[preset_selection, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=True) \
573
+ .then(fn=style_sorter.sort_styles, inputs=style_selections, outputs=style_selections, queue=False, show_progress=False) \
574
+
575
+ performance_selection.change(lambda x: [gr.update(interactive=not flags.Performance.has_restricted_features(x))] * 11 +
576
+ [gr.update(visible=not flags.Performance.has_restricted_features(x))] * 1 +
577
+ [gr.update(interactive=not flags.Performance.has_restricted_features(x), value=flags.Performance.has_restricted_features(x))] * 1,
578
+ inputs=performance_selection,
579
+ outputs=[
580
+ guidance_scale, sharpness, adm_scaler_end, adm_scaler_positive,
581
+ adm_scaler_negative, refiner_switch, refiner_model, sampler_name,
582
+ scheduler_name, adaptive_cfg, refiner_swap_method, negative_prompt, disable_intermediate_results
583
+ ], queue=False, show_progress=False)
584
+
585
+ output_format.input(lambda x: gr.update(output_format=x), inputs=output_format)
586
+
587
+ advanced_checkbox.change(lambda x: gr.update(visible=x), advanced_checkbox, advanced_column,
588
+ queue=False, show_progress=False) \
589
+ .then(fn=lambda: None, _js='refresh_grid_delayed', queue=False, show_progress=False)
590
+
591
+ def inpaint_mode_change(mode):
592
+ assert mode in modules.flags.inpaint_options
593
+
594
+ # inpaint_additional_prompt, outpaint_selections, example_inpaint_prompts,
595
+ # inpaint_disable_initial_latent, inpaint_engine,
596
+ # inpaint_strength, inpaint_respective_field
597
+
598
+ if mode == modules.flags.inpaint_option_detail:
599
+ return [
600
+ gr.update(visible=True), gr.update(visible=False, value=[]),
601
+ gr.Dataset.update(visible=True, samples=modules.config.example_inpaint_prompts),
602
+ False, 'None', 0.5, 0.0
603
+ ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
604
 
605
+ if mode == modules.flags.inpaint_option_modify:
606
  return [
607
+ gr.update(visible=True), gr.update(visible=False, value=[]),
608
  gr.Dataset.update(visible=False, samples=modules.config.example_inpaint_prompts),
609
+ True, modules.config.default_inpaint_engine_version, 1.0, 0.0
610
  ]
611
 
612
+ return [
613
+ gr.update(visible=False, value=''), gr.update(visible=True),
614
+ gr.Dataset.update(visible=False, samples=modules.config.example_inpaint_prompts),
615
+ False, modules.config.default_inpaint_engine_version, 1.0, 0.618
 
 
 
 
 
 
 
616
  ]
617
 
618
+ inpaint_mode.input(inpaint_mode_change, inputs=inpaint_mode, outputs=[
619
+ inpaint_additional_prompt, outpaint_selections, example_inpaint_prompts,
620
+ inpaint_disable_initial_latent, inpaint_engine,
621
+ inpaint_strength, inpaint_respective_field
622
+ ], show_progress=False, queue=False)
623
+
624
+ ctrls = [currentTask, generate_image_grid]
625
+ ctrls += [
626
+ prompt, negative_prompt, style_selections,
627
+ performance_selection, aspect_ratios_selection, image_number, output_format, image_seed,
628
+ read_wildcards_in_order, sharpness, guidance_scale
629
+ ]
630
+
631
+ ctrls += [base_model, refiner_model, refiner_switch] + lora_ctrls
632
+ ctrls += [input_image_checkbox, current_tab]
633
+ ctrls += [uov_method, uov_input_image]
634
+ ctrls += [outpaint_selections, inpaint_input_image, inpaint_additional_prompt, inpaint_mask_image]
635
+ ctrls += [disable_preview, disable_intermediate_results, disable_seed_increment]
636
+ ctrls += [adm_scaler_positive, adm_scaler_negative, adm_scaler_end, adaptive_cfg]
637
+ ctrls += [sampler_name, scheduler_name]
638
+ ctrls += [overwrite_step, overwrite_switch, overwrite_width, overwrite_height, overwrite_vary_strength]
639
+ ctrls += [overwrite_upscale_strength, mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint]
640
+ ctrls += [debugging_cn_preprocessor, skipping_cn_preprocessor, canny_low_threshold, canny_high_threshold]
641
+ ctrls += [refiner_swap_method, controlnet_softness]
642
+ ctrls += freeu_ctrls
643
+ ctrls += inpaint_ctrls
644
+
645
+ if not args_manager.args.disable_metadata:
646
+ ctrls += [save_metadata_to_images, metadata_scheme]
647
+
648
+ ctrls += ip_ctrls
649
+
650
+ def parse_meta(raw_prompt_txt, is_generating):
651
+ loaded_json = None
652
+ if is_json(raw_prompt_txt):
653
+ loaded_json = json.loads(raw_prompt_txt)
654
+
655
+ if loaded_json is None:
656
+ if is_generating:
657
+ return gr.update(), gr.update(), gr.update()
 
658
  else:
659
+ return gr.update(), gr.update(visible=True), gr.update(visible=False)
660
+
661
+ return json.dumps(loaded_json), gr.update(visible=False), gr.update(visible=True)
662
+
663
+ prompt.input(parse_meta, inputs=[prompt, state_is_generating], outputs=[prompt, generate_button, load_parameter_button], queue=False, show_progress=False)
664
+
665
+ load_parameter_button.click(modules.meta_parser.load_parameter_button_click, inputs=[prompt, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=False)
666
+
667
+ def trigger_metadata_import(filepath, state_is_generating):
668
+ parameters, metadata_scheme = modules.meta_parser.read_info_from_image(filepath)
669
+ if parameters is None:
670
+ print('Could not find metadata in the image!')
671
+ parsed_parameters = {}
672
+ else:
673
+ metadata_parser = modules.meta_parser.get_metadata_parser(metadata_scheme)
674
+ parsed_parameters = metadata_parser.parse_json(parameters)
675
+
676
+ return modules.meta_parser.load_parameter_button_click(parsed_parameters, state_is_generating)
677
+
678
+ metadata_import_button.click(trigger_metadata_import, inputs=[metadata_input_image, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=True) \
679
+ .then(style_sorter.sort_styles, inputs=style_selections, outputs=style_selections, queue=False, show_progress=False)
680
+
681
+ generate_button.click(lambda: (gr.update(visible=True, interactive=True), gr.update(visible=True, interactive=True), gr.update(visible=False, interactive=False), [], True),
682
+ outputs=[stop_button, skip_button, generate_button, gallery, state_is_generating]) \
683
+ .then(fn=refresh_seed, inputs=[seed_random, image_seed], outputs=image_seed) \
684
+ .then(fn=get_task, inputs=ctrls, outputs=currentTask) \
685
+ .then(fn=generate_clicked, inputs=currentTask, outputs=[progress_html, progress_window, progress_gallery, gallery]) \
686
+ .then(lambda: (gr.update(visible=True, interactive=True), gr.update(visible=False, interactive=False), gr.update(visible=False, interactive=False), False),
687
+ outputs=[generate_button, stop_button, skip_button, state_is_generating]) \
688
+ .then(fn=update_history_link, outputs=history_link) \
689
+ .then(fn=lambda: None, _js='playNotification').then(fn=lambda: None, _js='refresh_grid_delayed')
690
+
691
+ for notification_file in ['notification.ogg', 'notification.mp3']:
692
+ if os.path.exists(notification_file):
693
+ gr.Audio(interactive=False, value=notification_file, elem_id='audio_notification', visible=False)
694
+ break
695
+
696
+ def trigger_describe(mode, img):
697
+ if mode == flags.desc_type_photo:
698
+ from extras.interrogate import default_interrogator as default_interrogator_photo
699
+ return default_interrogator_photo(img), ["Fooocus V2", "Fooocus Enhance", "Fooocus Sharp"]
700
+ if mode == flags.desc_type_anime:
701
+ from extras.wd14tagger import default_interrogator as default_interrogator_anime
702
+ return default_interrogator_anime(img), ["Fooocus V2", "Fooocus Masterpiece"]
703
+ return mode, ["Fooocus V2"]
704
+
705
+ desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image],
706
+ outputs=[prompt, style_selections], show_progress=True, queue=True)
707
 
708
 
709
  def dump_default_english_config():