1580158
/
miaoshouai-assistant
mirror of https://github.com/miaoshouai/miaoshouai-assistant.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

v1.8 -download tab improvement, controlnet sdxl datasrouce update

pull/71/head
Superag 2023-10-25 12:39:22 +08:00
parent 1e071cd62f
commit 1b5bf91634
4 changed files with 507 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

500
configs/controlnet.json
View File

@ -685,6 +685,506 @@
},
"downloadUrl": "https://huggingface.co/CrucibleAI/ControlNetMediaPipeFace/resolve/main/control_v2p_sd21_mediapipe_face.pth",
"primary": false
},
{
"name": "ip-adapter_sd15_plus.pth",
"id": 34,
"sizeKB": 1581261.130859375,
"type": "Model",
"format": "Pth",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/ip-adapter_sd15_plus.pth",
"primary": false
},
{
"name": "ioclab_sd15_recolor.safetensors",
"id": 35,
"sizeKB": 5561261.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/ioclab_sd15_recolor.safetensors",
"primary": false
}
],
"images": [
{
"url": "https://msdn.miaoshouai.com/msai/kt/ez/9900922.png"
}
]
}
]
},
{
"id": 3,
"name": "Controlnet SDXL",
"description": "Collection of community SD control models for users to download flexibly.\n\nAll files are already float16 and in safetensor format.If you download the files from raw URL, you may need to rename them.\n\nHowever, files in https://huggingface.co/lllyasviel/sd_control_collection/tree/main are already renamed and can be directly downloaded.\n\nFeel free to contact us if you are author of any listed models and you want some models to be removed/added (by opening an issue in this HuggingFace page).",
"type": "Controlnet",
"poi": false,
"nsfw": false,
"allowNoCredit": true,
"allowCommercialUse": "Rent",
"allowDerivatives": true,
"allowDifferentLicense": true,
"stats": {
"downloadCount": 865,
"favoriteCount": 190,
"commentCount": 7,
"ratingCount": 2,
"rating": 5
},
"creator": {
"username": "lllyasviel",
"image": "https://huggingface.co/lllyasviel/ControlNet-v1-1"
},
"tags": [],
"modelVersions": [
{
"id": 301,
"modelId": 3,
"name": "Controlnet SDXL",
"createdAt": "2023-02-06T13:52:34.605Z",
"updatedAt": "2023-02-06T13:52:34.605Z",
"trainedWords": [],
"baseModel": "SDXL 1.0",
"earlyAccessTimeFrame": 0,
"description": "<p></p>",
"files": [
{
"name": "diffusers_xl_canny_mid.safetensors",
"id": 301,
"sizeKB": 556126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/diffusers_xl_canny_mid.safetensors",
"primary": false
},
{
"name": "diffusers_xl_depth_mid.safetensors",
"id": 302,
"sizeKB": 556126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/diffusers_xl_depth_mid.safetensors",
"primary": false
},
{
"name": "ip-adapter_xl.pth",
"id": 303,
"sizeKB": 556126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/diffusers_xl_depth_mid.safetensors",
"primary": false
},
{
"name": "kohya_controllllite_xl_blur.safetensors",
"id": 304,
"sizeKB": 46126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/kohya_controllllite_xl_blur.safetensors",
"primary": false
},
{
"name": "kohya_controllllite_xl_blur_anime.safetensors",
"id": 305,
"sizeKB": 46126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/kohya_controllllite_xl_blur_anime.safetensors",
"primary": false
},
{
"name": "kohya_controllllite_xl_canny.safetensors",
"id": 306,
"sizeKB": 46126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/kohya_controllllite_xl_canny.safetensors",
"primary": false
},
{
"name": "kohya_controllllite_xl_canny_anime.safetensors",
"id": 307,
"sizeKB": 46126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/kohya_controllllite_xl_canny_anime.safetensors",
"primary": false
},
{
"name": "kohya_controllllite_xl_depth.safetensors",
"id": 308,
"sizeKB": 46126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/kohya_controllllite_xl_depth.safetensors",
"primary": false
},
{
"name": "kohya_controllllite_xl_depth_anime.safetensors",
"id": 309,
"sizeKB": 46126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/kohya_controllllite_xl_depth_anime.safetensors",
"primary": false
},
{
"name": "kohya_controllllite_xl_openpose_anime_v2.safetensors",
"id": 310,
"sizeKB": 46126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/kohya_controllllite_xl_openpose_anime_v2.safetensors",
"primary": false
},
{
"name": "kohya_controllllite_xl_scribble_anime.safetensors",
"id": 311,
"sizeKB": 46126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/kohya_controllllite_xl_scribble_anime.safetensors",
"primary": false
},
{
"name": "sai_xl_canny_128lora.safetensors",
"id": 312,
"sizeKB": 410126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/sai_xl_canny_128lora.safetensors",
"primary": false
},
{
"name": "sai_xl_depth_128lora.safetensors",
"id": 313,
"sizeKB": 410126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/sai_xl_depth_128lora.safetensors",
"primary": false
},
{
"name": "sai_xl_recolor_128lora.safetensors",
"id": 314,
"sizeKB": 410126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/sai_xl_recolor_128lora.safetensors",
"primary": false
},
{
"name": "sai_xl_sketch_128lora.safetensors",
"id": 315,
"sizeKB": 410126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/sai_xl_sketch_128lora.safetensors",
"primary": false
},
{
"name": "t2i-adapter_diffusers_xl_canny.safetensors",
"id": 316,
"sizeKB": 158126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/t2i-adapter_diffusers_xl_canny.safetensors",
"primary": false
},
{
"name": "t2i-adapter_diffusers_xl_depth_midas.safetensors",
"id": 317,
"sizeKB": 158126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/t2i-adapter_diffusers_xl_depth_midas.safetensors",
"primary": false
},
{
"name": "t2i-adapter_diffusers_xl_depth_midas.safetensors",
"id": 318,
"sizeKB": 158126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/t2i-adapter_diffusers_xl_depth_midas.safetensors",
"primary": false
},
{
"name": "t2i-adapter_diffusers_xl_depth_zoe.safetensors",
"id": 319,
"sizeKB": 158126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/t2i-adapter_diffusers_xl_depth_zoe.safetensors",
"primary": false
},
{
"name": "t2i-adapter_diffusers_xl_lineart.safetensors",
"id": 320,
"sizeKB": 158126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/t2i-adapter_diffusers_xl_lineart.safetensors",
"primary": false
},
{
"name": "t2i-adapter_diffusers_xl_openpose.safetensors",
"id": 321,
"sizeKB": 158126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/t2i-adapter_diffusers_xl_openpose.safetensors",
"primary": false
},
{
"name": "t2i-adapter_diffusers_xl_sketch.safetensors",
"id": 322,
"sizeKB": 158126.130859375,
"type": "Model",
"format": "safetensors",
"pickleScanResult": "Success",
"pickleScanMessage": "No Pickle imports",
"virusScanResult": "Success",
"scannedAt": "2023-02-06T13:55:58.281Z",
"hashes": {
"AutoV2": "833E816807",
"SHA256": "833E816807E8D3658EBC65436908F690603A8E0E8D26143A90050F32958C9E4E",
"CRC32": "87F68C12",
"BLAKE3": "7A28D3D22520D61B58338791522A14B7B33A6A9CE552FF7D68D7E2688CAB3266"
},
"downloadUrl": "https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/t2i-adapter_diffusers_xl_sketch.safetensors",
"primary": false
}
],
"images": [

2
configs/hugging_face.json
View File

@ -26,7 +26,7 @@
{
"id": 1,
"modelId": 7019,
"name": "Controlnet 1.0",
"name": "Neg Embedding",
"createdAt": "2023-02-06T13:52:34.605Z",
"updatedAt": "2023-02-06T13:52:34.605Z",
"trainedWords": [],

4
scripts/assistant/miaoshou.py
View File

@ -471,7 +471,6 @@ class MiaoShouAssistant(object):
if key in item['stats']:
return item['stats'][key]
print(sort_by)
self.runtime.model_source = new_model_source
show_download_button = self.runtime.model_source != "liandange.com"
@ -480,6 +479,7 @@ class MiaoShouAssistant(object):
else:
self.runtime.sorted_model_set = self.runtime.model_set
print(len(self.runtime.sorted_model_set))
if self.runtime.ds_models:
self.runtime.ds_models.samples = self.runtime.sorted_model_set
self.runtime.ds_models.update(samples=self.runtime.sorted_model_set)
@ -488,7 +488,7 @@ class MiaoShouAssistant(object):
images = self.runtime.get_images_html(search, chk_nsfw, base_model, model_type, model_tag)
#images = self.runtime.sort_dataset(search, chk_nsfw, base_model, model_type, model_tag)
print(images[0])
self.runtime.ds_models.samples = images
self.runtime.ds_models.update(samples=self.runtime.sorted_model_set)

8
scripts/runtime/msai_runtime.py
View File

@ -197,7 +197,7 @@ class MiaoshouRuntime(object):
def get_images_html(self, search: str = '', chk_nsfw: bool = False, base_model=None, model_type: str = 'All', model_tag: str = 'All') -> t.List[str]:
if base_model is None:
base_model = ['SD 1.5']
base_model = []
self.logger.info(f"get_image_html: model_type = {model_type}, and search pattern = '{search}'")
model_cover_thumbnails = []
@ -230,7 +230,7 @@ class MiaoshouRuntime(object):
self._allow_nsfw = chk_nsfw
if (model_type == 'All' or model_type in model.get('type')) \
and (self.allow_nsfw or (not self.allow_nsfw and not model.get('nsfw'))) \
and ('baseModel' not in model['modelVersions'][0].keys() or (base_model is not None and model['modelVersions'][0]['baseModel'] in base_model)) \
and ('baseModel' not in model['modelVersions'][0].keys() or (len(base_model) > 0 and model['modelVersions'][0]['baseModel'] in base_model)) \
and (model_tag == 'All' or model_tag.lower() in model.get('tags')):
model_cover_thumbnails.append([
@ -518,7 +518,7 @@ class MiaoshouRuntime(object):
def set_nsfw(self, search='', nsfw_checker=False, base_model=None, model_type='All', model_tag='All') -> t.Dict:
if base_model is None:
base_model = ['SD 1.5']
base_model = []
self._allow_nsfw = nsfw_checker
new_list = self.get_images_html(search, nsfw_checker, base_model, model_type, model_tag)
if self._ds_models is None:
@ -539,7 +539,6 @@ class MiaoshouRuntime(object):
self.logger.error(f"_ds_models is not initialized")
return {}
print(base_model)
new_list = self.get_images_html(search, chk_nsfw, base_model, model_type, model_tag)
self._ds_models.samples = new_list
@ -652,6 +651,7 @@ class MiaoshouRuntime(object):
htmlDetail = "<div><p>No info found</p></div>"
mid = models[1]
print(mid)
# TODO: use map to enhance the performances
if self.active_model_set == 'model_set':