k-l-lambda commited on
Commit
ac592b3
1 Parent(s): 1b951b8

app.py: added lora models.

Browse files
Files changed (1) hide show
  1. app.py +40 -9
app.py CHANGED
@@ -33,18 +33,18 @@ controlnet_depth_model = 'diffusers/controlnet-depth-sdxl-1.0-small'
33
  SDXL_MODELS = [
34
  "albedobaseXL_v04_130099",
35
  "altxl_v60_146691",
 
36
  "animeArtDiffusionXL_alpha2_91872",
37
  "animeArtDiffusionXL_alpha3_93120",
38
  "animeIllustDiffusion_v04_117809",
39
- "animagineXLV3_v30_231047",
40
  "breakdomainxl_V05g_124265",
41
  "brixlAMustInYour_v40Dagobah_145992",
42
  "cinemaxAlphaSDXLCinema_alpha1_107473",
43
  "cineroXLPhotomatic_v12aPHENO_137703",
44
  "clearhungAnimeXL_v10_117716",
45
  "copaxTimelessxlSDXL1_colorfulV2_100729",
46
- "counterfeitxl__98184",
47
  "counterfeitxl_v10_108721",
 
48
  "crystalClearXL_ccxl_97637",
49
  "dreamshaperXL09Alpha_alpha2Xl10_91562",
50
  "dynavisionXLAllInOneStylized_alpha036FP16Bakedvae_99980",
@@ -53,9 +53,9 @@ SDXL_MODELS = [
53
  "fenrisxl_145_134980",
54
  "foddaxlPhotorealism_v45_122788",
55
  "formulaxl_v10_104889",
 
56
  "juggernautXL_version2_113240",
57
  "juggernautXL_version5_126522",
58
- "juggernautXL_v8Rundiffusion_227002",
59
  "kohakuXL_alpha7_111843",
60
  "LahMysteriousSDXL_v40_122478",
61
  "leosamsHelloworldSDXLModel_helloworldSDXL10_112178",
@@ -78,17 +78,14 @@ SDXL_MODELS = [
78
  "realvisxlV20_v20Bakedvae_129156",
79
  "riotDiffusionXL_v20_139293",
80
  "roxl_v10_109354",
81
- "sd_xl_base_0.9",
82
- "sd_xl_base_1.0",
83
  "sdxlNijiSpecial_sdxlNijiSE_115638",
84
  "sdxlNijiV3_sdxlNijiV3_104571",
85
  "sdxlNijiV51_sdxlNijiV51_112807",
86
- "sd_xl_refiner_1.0",
87
  "sdxlUnstableDiffusers_v8HEAVENSWRATH_133813",
88
- "sdXL_v10Refiner_91495",
89
  "sdxlYamersAnimeUltra_yamersAnimeV3_121537",
 
 
90
  "shikianimexl_v10_93788",
91
- #"stable-diffusion-xl-base-1.0",
92
  "theTalosProject_v10_117893",
93
  "thinkdiffusionxl_v10_145931",
94
  "voidnoisecorexl_r1486_150780",
@@ -97,9 +94,31 @@ SDXL_MODELS = [
97
  "xl13AsmodeusSFWNSFW_v22BakedVAE_111954",
98
  "xxmix9realisticsdxl_v10_123235",
99
  "zavychromaxl_b2_103298",
100
- "zavychromaxl_v21_129006",
101
  ]
102
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
103
  CONTROLNET_DICT = dict(
104
  pose={
105
  'model_name': 'controlnet-openpose-sdxl-1.0',
@@ -236,6 +255,7 @@ def upload_assets_with_cache (client, paths):
236
  def generate_image (
237
  novita_key1,
238
  model_name,
 
239
  face_image_path,
240
  pose_image_path,
241
  prompt,
@@ -303,6 +323,10 @@ def generate_image (
303
  'controlnet': {
304
  'units': [CONTROLNET_DICT[name] for name in controlnet_selection if name in CONTROLNET_DICT],
305
  },
 
 
 
 
306
  'image_num': 1,
307
  'steps': num_steps,
308
  'seed': seed,
@@ -415,6 +439,12 @@ with gr.Blocks(css=css) as demo:
415
  value=DEFAULT_MODEL_NAME,
416
  )
417
 
 
 
 
 
 
 
418
  style = gr.Dropdown(
419
  label='Style template',
420
  choices=STYLE_NAMES,
@@ -528,6 +558,7 @@ with gr.Blocks(css=css) as demo:
528
  inputs=[
529
  novita_key,
530
  model_name,
 
531
  face_file,
532
  pose_file,
533
  prompt,
 
33
  SDXL_MODELS = [
34
  "albedobaseXL_v04_130099",
35
  "altxl_v60_146691",
36
+ "animagineXLV3_v30_231047",
37
  "animeArtDiffusionXL_alpha2_91872",
38
  "animeArtDiffusionXL_alpha3_93120",
39
  "animeIllustDiffusion_v04_117809",
 
40
  "breakdomainxl_V05g_124265",
41
  "brixlAMustInYour_v40Dagobah_145992",
42
  "cinemaxAlphaSDXLCinema_alpha1_107473",
43
  "cineroXLPhotomatic_v12aPHENO_137703",
44
  "clearhungAnimeXL_v10_117716",
45
  "copaxTimelessxlSDXL1_colorfulV2_100729",
 
46
  "counterfeitxl_v10_108721",
47
+ "counterfeitxl__98184",
48
  "crystalClearXL_ccxl_97637",
49
  "dreamshaperXL09Alpha_alpha2Xl10_91562",
50
  "dynavisionXLAllInOneStylized_alpha036FP16Bakedvae_99980",
 
53
  "fenrisxl_145_134980",
54
  "foddaxlPhotorealism_v45_122788",
55
  "formulaxl_v10_104889",
56
+ "juggernautXL_v8Rundiffusion_227002",
57
  "juggernautXL_version2_113240",
58
  "juggernautXL_version5_126522",
 
59
  "kohakuXL_alpha7_111843",
60
  "LahMysteriousSDXL_v40_122478",
61
  "leosamsHelloworldSDXLModel_helloworldSDXL10_112178",
 
78
  "realvisxlV20_v20Bakedvae_129156",
79
  "riotDiffusionXL_v20_139293",
80
  "roxl_v10_109354",
 
 
81
  "sdxlNijiSpecial_sdxlNijiSE_115638",
82
  "sdxlNijiV3_sdxlNijiV3_104571",
83
  "sdxlNijiV51_sdxlNijiV51_112807",
 
84
  "sdxlUnstableDiffusers_v8HEAVENSWRATH_133813",
 
85
  "sdxlYamersAnimeUltra_yamersAnimeV3_121537",
86
+ "sd_xl_base_0.9",
87
+ "sd_xl_base_1.0",
88
  "shikianimexl_v10_93788",
 
89
  "theTalosProject_v10_117893",
90
  "thinkdiffusionxl_v10_145931",
91
  "voidnoisecorexl_r1486_150780",
 
94
  "xl13AsmodeusSFWNSFW_v22BakedVAE_111954",
95
  "xxmix9realisticsdxl_v10_123235",
96
  "zavychromaxl_b2_103298",
 
97
  ]
98
 
99
+ LORA_MODELS = [
100
+ "DI_belle_delphine_sdxl_v1_93586",
101
+ #"NsfwPovAllInOneLoraSdxl-000009MINI_120545",
102
+ "NsfwPovAllInOneLoraSdxl-000009_120561",
103
+ "acidzlime-sdxl_154149",
104
+ "add-detail-xl_99264",
105
+ "bwporcelaincd_xl-000007_124344",
106
+ "concept_pov_dt_xl2-000020_119643",
107
+ "epoxy_skull-sdxl_153213",
108
+ "landscape-painting-sdxl_v2_111037",
109
+ "polyhedron_all_sdxl-000004_110557",
110
+ "ral-beer-sdxl_235173",
111
+ "ral-wtchz-sdxl_233487",
112
+ "sdxl_cute_social_comic-000002_107980",
113
+ "sdxl_glass_136034",
114
+ "sdxl_lightning_8step_lora_290441",
115
+ "sdxl_offset_example_v10_113006",
116
+ "sdxl_wrong_lora",
117
+ "xl_more_art-full_v1_113467",
118
+ "xl_yoshiaki_kawajiri_v1r64_126468",
119
+ ]
120
+
121
+
122
  CONTROLNET_DICT = dict(
123
  pose={
124
  'model_name': 'controlnet-openpose-sdxl-1.0',
 
255
  def generate_image (
256
  novita_key1,
257
  model_name,
258
+ lora_selection,
259
  face_image_path,
260
  pose_image_path,
261
  prompt,
 
323
  'controlnet': {
324
  'units': [CONTROLNET_DICT[name] for name in controlnet_selection if name in CONTROLNET_DICT],
325
  },
326
+ 'loras': [dict(
327
+ model_name=f'{name}.safetensors',
328
+ scale=1,
329
+ ) for name in lora_selection],
330
  'image_num': 1,
331
  'steps': num_steps,
332
  'seed': seed,
 
439
  value=DEFAULT_MODEL_NAME,
440
  )
441
 
442
+ with gr.Accordion('Lora', open=False):
443
+ lora_selection = gr.CheckboxGroup(
444
+ LORA_MODELS, value=[],
445
+ info='Try lora models mix in generation'
446
+ )
447
+
448
  style = gr.Dropdown(
449
  label='Style template',
450
  choices=STYLE_NAMES,
 
558
  inputs=[
559
  novita_key,
560
  model_name,
561
+ lora_selection,
562
  face_file,
563
  pose_file,
564
  prompt,