app.py: added lora models.

app.py

@@ -33,18 +33,18 @@ controlnet_depth_model = 'diffusers/controlnet-depth-sdxl-1.0-small'
 SDXL_MODELS = [
 	"albedobaseXL_v04_130099",
 	"altxl_v60_146691",
+	"animagineXLV3_v30_231047",
 	"animeArtDiffusionXL_alpha2_91872",
 	"animeArtDiffusionXL_alpha3_93120",
 	"animeIllustDiffusion_v04_117809",
-	"animagineXLV3_v30_231047",
 	"breakdomainxl_V05g_124265",
 	"brixlAMustInYour_v40Dagobah_145992",
 	"cinemaxAlphaSDXLCinema_alpha1_107473",
 	"cineroXLPhotomatic_v12aPHENO_137703",
 	"clearhungAnimeXL_v10_117716",
 	"copaxTimelessxlSDXL1_colorfulV2_100729",
-	"counterfeitxl__98184",
 	"counterfeitxl_v10_108721",
+	"counterfeitxl__98184",
 	"crystalClearXL_ccxl_97637",
 	"dreamshaperXL09Alpha_alpha2Xl10_91562",
 	"dynavisionXLAllInOneStylized_alpha036FP16Bakedvae_99980",
@@ -53,9 +53,9 @@ SDXL_MODELS = [
 	"fenrisxl_145_134980",
 	"foddaxlPhotorealism_v45_122788",
 	"formulaxl_v10_104889",
+	"juggernautXL_v8Rundiffusion_227002",
 	"juggernautXL_version2_113240",
 	"juggernautXL_version5_126522",
-	"juggernautXL_v8Rundiffusion_227002",
 	"kohakuXL_alpha7_111843",
 	"LahMysteriousSDXL_v40_122478",
 	"leosamsHelloworldSDXLModel_helloworldSDXL10_112178",
@@ -78,17 +78,14 @@ SDXL_MODELS = [
 	"realvisxlV20_v20Bakedvae_129156",
 	"riotDiffusionXL_v20_139293",
 	"roxl_v10_109354",
-	"sd_xl_base_0.9",
-	"sd_xl_base_1.0",
 	"sdxlNijiSpecial_sdxlNijiSE_115638",
 	"sdxlNijiV3_sdxlNijiV3_104571",
 	"sdxlNijiV51_sdxlNijiV51_112807",
-	"sd_xl_refiner_1.0",
 	"sdxlUnstableDiffusers_v8HEAVENSWRATH_133813",
-	"sdXL_v10Refiner_91495",
 	"sdxlYamersAnimeUltra_yamersAnimeV3_121537",
+	"sd_xl_base_0.9",
+	"sd_xl_base_1.0",
 	"shikianimexl_v10_93788",
-	#"stable-diffusion-xl-base-1.0",
 	"theTalosProject_v10_117893",
 	"thinkdiffusionxl_v10_145931",
 	"voidnoisecorexl_r1486_150780",
@@ -97,9 +94,31 @@ SDXL_MODELS = [
 	"xl13AsmodeusSFWNSFW_v22BakedVAE_111954",
 	"xxmix9realisticsdxl_v10_123235",
 	"zavychromaxl_b2_103298",
-	"zavychromaxl_v21_129006",
 ]
 
+LORA_MODELS = [
+	"DI_belle_delphine_sdxl_v1_93586",
+	#"NsfwPovAllInOneLoraSdxl-000009MINI_120545",
+	"NsfwPovAllInOneLoraSdxl-000009_120561",
+	"acidzlime-sdxl_154149",
+	"add-detail-xl_99264",
+	"bwporcelaincd_xl-000007_124344",
+	"concept_pov_dt_xl2-000020_119643",
+	"epoxy_skull-sdxl_153213",
+	"landscape-painting-sdxl_v2_111037",
+	"polyhedron_all_sdxl-000004_110557",
+	"ral-beer-sdxl_235173",
+	"ral-wtchz-sdxl_233487",
+	"sdxl_cute_social_comic-000002_107980",
+	"sdxl_glass_136034",
+	"sdxl_lightning_8step_lora_290441",
+	"sdxl_offset_example_v10_113006",
+	"sdxl_wrong_lora",
+	"xl_more_art-full_v1_113467",
+	"xl_yoshiaki_kawajiri_v1r64_126468",
+]
+
+
 CONTROLNET_DICT = dict(
 	pose={
 		'model_name': 'controlnet-openpose-sdxl-1.0',
@@ -236,6 +255,7 @@ def upload_assets_with_cache (client, paths):
 def generate_image (
 	novita_key1,
 	model_name,
+	lora_selection,
 	face_image_path,
 	pose_image_path,
 	prompt,
@@ -303,6 +323,10 @@ def generate_image (
 			'controlnet': {
 				'units': [CONTROLNET_DICT[name] for name in controlnet_selection if name in CONTROLNET_DICT],
 			},
+			'loras': [dict(
+				model_name=f'{name}.safetensors',
+				scale=1,
+			) for name in lora_selection],
 			'image_num': 1,
 			'steps': num_steps,
 			'seed': seed,
@@ -415,6 +439,12 @@ with gr.Blocks(css=css) as demo:
 				value=DEFAULT_MODEL_NAME,
 			)
 
+			with gr.Accordion('Lora', open=False):
+				lora_selection = gr.CheckboxGroup(
+					LORA_MODELS, value=[],
+					info='Try lora models mix in generation'
+				)
+
 			style = gr.Dropdown(
 				label='Style template',
 				choices=STYLE_NAMES,
@@ -528,6 +558,7 @@ with gr.Blocks(css=css) as demo:
 			inputs=[
 				novita_key,
 				model_name,
+				lora_selection,
 				face_file,
 				pose_file,
 				prompt,