1# SPDX-License-Identifier: MPL-2.0 AND LicenseRef-Commons-Clause-License-Condition-1.0
- 2# <!-- // /* d a r k s h a p e s */ -->
- 3
- 4"""自動化索引"""
- 5# regex to remove \[[^\]]*\]
- 6# (?m)^\s*"[^"]+"(?=\s*:)
- 7# (?m)^\s*"[^"]+"\s?:
- 8# modelspec sai https://github.com/Stability-AI/ModelSpec
- 9
- 10from importlib import import_module
- 11from typing import Dict, List, Tuple
- 12import re
- 13import torch
- 14from diffusers import _import_structure
- 15
- 16from mir.maid import MIRDatabase
- 17from mir.spec.mir import mir_entry
- 18from mir.tag import make_mir_tag, make_scheduler_tag, tag_pipe, tag_base_model
- 19from mir.config.logging import nfo, dbuq
- 20from mir.config.conversion import slice_number
- 21from mir.indexers import diffusers_index, transformers_index
- 22
- 23
- 24sd1_series, sd1_comp = make_mir_tag("stable-diffusion-v1-5/stable-diffusion-v1-5")
- 25sdxl_series, sdxl_comp = make_mir_tag("stabilityai/stable-diffusion-xl-base-1.0")
- 26dev_series, dev_comp = make_mir_tag("black-forest-labs/FLUX.1-dev")
- 27schnell_series, schnell_comp = make_mir_tag("black-forest-labs/FLUX.1-schnell")
- 28ssd_series, ssd_comp = make_mir_tag("segmind/SSD-1B")
- 29vega_series, vega_comp = make_mir_tag("segmind/Segmind-Vega")
- 30sd3_series, sd3_comp = make_mir_tag("stable-diffusion-3.5-medium") #
- 31
- 32# def gen_attention_processors(mir_db: MIRDatabase): # upstream not quite ready for this yet
- 33# from diffusers.models.attention_processor import AttentionProcessor
- 34
- 35# mir_data
- 36# for series, comp_name in mir_data.items():
- 37# id_segment = series.split(".")
- 38# for compatibility in comp_name:
- 39# dbug(id_segment)
- 40# try:
- 41# mir_db.add(
- 42# mir_entry(
- 43# domain=id_segment[0],
- 44# arch=id_segment[1],
- 45# series=id_segment[2],
- 46# comp=compatibility,
- 47# **mir_data[series][compatibility],
- 48# ),
- 49# )
- 50# except IndexError as error_log:
- 51# nfo(f"Failed to create series: {series} compatibility: {comp_name} ")
- 52# dbug(error_log)
- 53
- 54
- 55# def gen_guiders(mir_db: MIRDatabase): # upstream not quite ready for this yet
- 56# from nnll.metadata.helpers import snake_caseify
- 57# from diffusers.guider import GuiderType
- 58
- 59# guider_type = GuiderType
- 60# for comp_name in guider_type.items():
- 61# class_obj = comp_name.__name__
- 62# mir_data = {"pkg": {0: {"diffusers": class_obj}}}
- 63# try:
- 64# mir_db.add(
- 65# mir_entry(
- 66# domain="ops",
- 67# arch="noise_prediction",
- 68# series="guider",
- 69# comp=snake_caseify(class_obj),
- 70# **mir_data,
- 71# ),
- 72# )
- 73# except IndexError as error_log:
- 74# nfo(f"Failed to create compatibility: {class_obj}")
- 75# dbug(error_log)
- 76
- 77
- 78# (
- 79# "info.unet",
- 80# "stable-cascade",
- 81# {
- 82# "combined": {
- 83# "pkg": {
- 84# 0: { # decoder=decoder_unet
- 85# "precision": "ops.precision.bfloat.B16",
- 86# "generation": {
- 87# "negative_prompt": "",
- 88# "num_inference_steps": 20,
- 89# "guidance_scale": 4.0,
- 90# "num_images_per_prompt": 1,
- 91# "width": 1024,
- 92# "height": 1024,
- 93# },
- 94# },
- 95# "pkg_alt": {
- 96# 0: {
- 97# "diffusers": {
- 98# "StableCascadeCombinedPipeline": {
- 99# "negative_prompt": "",
- 100# "num_inference_steps": 10,
- 101# "prior_num_inference_steps": 20,
- 102# "prior_guidance_scale": 3.0,
- 103# }
- 104# },
- 105# }
- 106# },
- 107# }
- 108# }
- 109# },
- 110# ),
- 111
- 112
- 113def assimilate(mir_db: MIRDatabase, data_tuple: List[Tuple[Dict[str, any]]]) -> None:
- 114 """Merge new data into a pre-generated MIR database, updating while preserving existing data structures.\n
- 115 :param mir_db: The MIRDatabase instance
- 116 :param data_tuple: A list of tuples, each containing:\n
- 117 - arch (str): The architecture name
- 118 - series (str): The series name
- 119 - `new_data`: New data to be merged into the database.
- 120 :raises TypeError: If any field in `new_data` is not a dictionary.
- 121 """
- 122
- 123 def update_nested_dict(target, source):
- 124 for key, value in source.items():
- 125 if isinstance(value, dict) and key in target:
- 126 if isinstance(target, dict):
- 127 update_nested_dict(target[key], value)
- 128 else:
- 129 if isinstance(source, dict):
- 130 # dbuq(target)
- 131 target.setdefault(key, value)
- 132 else:
- 133 target = {key: value}
- 134
- 135 dbuq(f"{data_tuple}, {len(data_tuple)}")
- 136 for arch, series, new_data in data_tuple:
- 137 mir_data = mir_db.database[f"{arch}.{series}"]
- 138 for comp, field_data in new_data.items():
- 139 if not isinstance(field_data, dict):
- 140 raise TypeError(f"{field_data} <-- Cannot combine with database: Not `dict()`")
- 141
- 142 # dbuq(f"{arch}.{series} : {comp}")
- 143 update_nested_dict(mir_data.setdefault(comp, {}), field_data)
- 144
- 145 if series == sdxl_series:
- 146 for field, field_data in field_data.items():
- 147 if isinstance(field_data, dict):
- 148 for definition, sub_def_data in field_data.items():
- 149 # dbug(definition)
- 150 if isinstance(sub_def_data, dict):
- 151 mir_data[comp][field].setdefault(definition, {})
- 152 update_nested_dict(mir_data[comp][field][definition], sub_def_data)
- 153
- 154
- 155def hf_pkg_to_mir(mir_db: MIRDatabase):
- 156 """Generate MIR HF Hub model database"""
- 157 mir_data = diffusers_index() | transformers_index()
- 158 for series, comp_name in mir_data.items():
- 159 id_segment = series.split(".")
- 160 for compatibility in comp_name:
- 161 # dbug(id_segment)
- 162 try:
- 163 mir_db.add(
- 164 mir_entry(
- 165 domain=id_segment[0],
- 166 arch=id_segment[1],
- 167 series=id_segment[2],
- 168 comp=compatibility,
- 169 **mir_data[series][compatibility],
- 170 ),
- 171 )
- 172 except IndexError: # as error_log:
- 173 nfo(f"Failed to create series: {series} compatibility: {comp_name} ")
- 174 # dbug(error_log)
- 175
- 176
- 177def add_mir_dtype(mir_db: MIRDatabase):
- 178 """Create mir info database"""
- 179
- 180 available_dtypes: List[str] = [dtype for dtype in torch.__dict__.values() if isinstance(dtype, torch.dtype)]
- 181 series_name = "_"
- 182 for precision in available_dtypes:
- 183 dep_name, class_name = str(precision).split(".")
- 184 if "_" in class_name:
- 185 comp_name = class_name[0].upper() + "8_" + class_name.split("_")[1].upper()
- 186 if comp_name.endswith("FN"):
- 187 comp_name = comp_name[:-2]
- 188 else:
- 189 comp_name = class_name[0].upper() + str(slice_number(class_name))
- 190 variant_name = class_name.replace("bfloat", "bf").replace("float", "fp")
- 191 dbuq(variant_name)
- 192 patterns = [r"complex", r"bits", r"quint", r"uint", r"int", r"bfloat", r"float", r"bool"]
- 193 for precision_name in patterns:
- 194 compiled = re.compile(precision_name)
- 195 dtype = re.search(compiled, class_name)
- 196 if dtype:
- 197 series_name = dtype.group()
- 198 break
- 199
- 200 mir_db.add(
- 201 mir_entry(
- 202 domain="ops",
- 203 arch="precision",
- 204 series=series_name,
- 205 comp=comp_name,
- 206 pkg={0: {dep_name.lower(): {class_name.lower(): {"variant": variant_name}}}},
- 207 )
- 208 )
- 209
- 210
- 211def add_mir_schedulers(mir_db: MIRDatabase):
- 212 """Create mir info database"""
- 213
- 214 for class_name in _import_structure["schedulers"]:
- 215 if class_name != "SchedulerMixin":
- 216 series_name, comp_name = make_scheduler_tag(class_name)
- 217 class_obj = import_module("diffusers.schedulers")
- 218 class_path = getattr(class_obj, class_name).__module__
- 219 mir_db.add(
- 220 mir_entry(
- 221 domain="ops",
- 222 arch="scheduler",
- 223 series=series_name,
- 224 comp=comp_name.lower(),
- 225 pkg={
- 226 0: {
- 227 "diffusers": class_name,
- 228 "module_path": class_path,
- 229 },
- 230 },
- 231 )
- 232 )
- 233
- 234 class_name = "KarrasDiffusionSchedulers"
- 235 series_name, comp_name = make_scheduler_tag(class_name)
- 236 class_obj = import_module("diffusers.schedulers.scheduling_utils")
- 237 class_path = getattr(class_obj, class_name).__module__
- 238 mir_db.add(
- 239 mir_entry(
- 240 domain="ops",
- 241 arch="scheduler",
- 242 series=series_name,
- 243 comp=comp_name,
- 244 pkg={
- 245 0: {
- 246 "diffusers": class_name,
- 247 "module_path": class_path,
- 248 },
- 249 },
- 250 ),
- 251 )
- 252
- 253
- 254# def auto_gan etc etc
- 255# ai-forever/Real-ESRGAN
- 256
- 257
- 258def mir_update(mir_db: MIRDatabase, task_list: list = None, pipe_list: list = None):
- 259 """Create mir unet info database"""
- 260
- 261 diffusers_addons = [
- 262 (
- 263 "stabilityai/stable-diffusion-xl-base-1.0",
- 264 "StableDiffusionXLPipeline",
- 265 {
- 266 "pkg": {
- 267 0: {
- 268 "precision": "ops.precision.float.F16",
- 269 "generation": {
- 270 "denoising_end": 0.8,
- 271 "num_inference_steps": 40,
- 272 "output_type": "latent",
- 273 "safety_checker": False,
- 274 "width": 1024,
- 275 "height": 1024,
- 276 },
- 277 },
- 278 1: {"diffusers": "DiffusionPipeline"},
- 279 },
- 280 "file_256": [
- 281 "357650fbfb3c7b4d94c1f5fd7664da819ad1ff5a839430484b4ec422d03f710a", # diffusers
- 282 "83e012a805b84c7ca28e5646747c90a243c65c8ba4f070e2d7ddc9d74661e139", # fp16 diffusers
- 283 "31e35c80fc4829d14f90153f4c74cd59c90b779f6afe05a74cd6120b893f7e5b", # modelspec sai
- 284 "6f001c090fb13c0d0f8b0a5916da814712a94400b99471fabe77c1c4a51ecaaf", # onnx
- 285 ],
- 286 "layer_256": [
- 287 "62a5ab1b5fdfa4fedb32323841298c6effe1af25be94a8583350b0a7641503ef", # any modelspec sai
- 288 "34dff8d98898baa0f10e71943e56b588cc114253b0d2f1051f3ce7a8a45fee0b", # diffusers
- 289 "56b1ccd89b0d6ab658048aa34d659788b6ed663f13ef566f4b11bccef590b9da", # diffusers fp16
- 290 ],
- 291 "layer_b3": [
- 292 "8be44fa13c1efa60f8bcadaa57f1d718473f9660f03c4f0e65dc037960d8cba1", # any modelspec sai
- 293 "c9ab95ed1851418b65ef99651c1eb6bbdd2e3b0715e0e435d6d1e56ce310fac3", # diffusers
- 294 "adfa260098d87616d748e3cf9c10bb2c90ff8890a84abbb2853d4aa69664070b", # diffusers fp16
- 295 ],
- 296 "identifiers": ["logit_scale", "conditioner.embedders.0.transformer.text_model.encoder.layers.0.self_attn.k_proj.weight", "add_embedding.linear_2.bias"],
- 297 },
- 298 ),
- 299 (
- 300 "stabilityai/stable-diffusion-xl-refiner-1.0",
- 301 "StableDiffusionXLImg2ImgPipeline",
- 302 {
- 303 "pkg": {
- 304 1: {
- 305 "diffusers": "DiffusionPipeline",
- 306 "generation": {"num_inference_steps": 40, "denoising_end": 0.8},
- 307 }
- 308 },
- 309 "identifiers": ["conditioner.embedders.0.model.transformer.resblocks.9.mlp.c_proj.bias"],
- 310 "file_256": [
- 311 "54f9cd2f2daf3aeec0b2708fa3dbc0e84e4f8ddd1ddead42e5bc60c6572c989f", # diffusers
- 312 "7440042bbdc8a24813002c09b6b69b64dc90fded4472613437b7f55f9b7d9c5f", # modelspec sai
- 313 "3ea0376dcf065eaefd27806394a90e310001b1a71d4f1cf1f655e86c0e566ffe", # fp16 diffusers
- 314 ],
- 315 "layer_b3": [
- 316 "6281355dbb37e5769c9460ae0ac75506d89932e2f97b09d9ade32ecf191e75ba",
- 317 "afb0639aae2eb65577c12d4a30cf7c9b3620ae63ba64a8fa632b58608c8a7a2e",
- 318 "669046014b69d98ab0f6fbb59547644436e0275f8b638f467ce2a873c3313683",
- 319 ],
- 320 "layer_256": [
- 321 "bb9eadbfabb52c0d8645783525a3fa70b59e9d7d09d5290d742a303262e793a2",
- 322 "c5adb56fe51343af2c3d493eb9f41515c204bd91eb9f40b983d45f70a1fa3b6d",
- 323 "1f838e39ed6e916258aee6990b72c09b34aa8eb3b5342234a497b8852b3df1c6",
- 324 ],
- 325 },
- 326 ),
- 327 (
- 328 "lodestones/Chroma",
- 329 "ChromaPipeline",
- 330 {
- 331 "pkg": {
- 332 1: {
- 333 "generation": {"neg_text": "", "num_steps": "28", "latent_size": [64, 64]},
- 334 }
- 335 },
- 336 "file_256": [
- 337 "53adcb3b6b6005758d40e2d8058b044ed4892bc8616efb7a62cc2dd384be07de", # v1
- 338 "2c41e8a9831f3be1eaff2c2ed590abb62e4534e814f7ec58a5fd74ff71dc2036", # v46,
- 339 "0a7b2d9699dbd22b3744ee2692900cabcfb731a43dac13729c33807f2bb7c9f6", # v37 detail
- 340 "6ddc9e2bbe3376ab5ee9f10b2d947f127b6bf6f879f06f316a2208bb0da357b8", # mlx chroma / v36 detail
- 341 ],
- 342 "layer_b3": [
- 343 "15e227ced8a89c41abaa9cc44f84dfffdf5ead0c626035e5a2dde2bbb0935479",
- 344 ],
- 345 "layer_256": ["a4daa6ff6f45ca70c738adb8c19bc3b6f228df931e6bf2a3394463e4dd7ec882"],
- 346 },
- 347 ),
- 348 (
- 349 "fal/AuraFlow",
- 350 "AuraFlowPipeline",
- 351 {
- 352 "identifiers": [[8192, 3072], "mlpX.c_fc2.weight", "joint_transformer_blocks.2.ff_context.linear_2.weight"],
- 353 "file_256": [
- 354 "ce3e475246258b94ee9dcb8b83292cb34edfffc2bbde46c74604d9c6cd7c585c",
- 355 "526be97cf581c89ad87c6b19c1f7c2378851137698f7ec436596d061a382d37b", # sai
- 356 "6a40b011f287452dbca80face78e667055904c5ad97eb2097ade3200259b2203", # diffusers fp16
- 357 "05e5493018333d947bb5940083dbc2f071093027ff414bc5b1b1229e4836e5cb", # diffusers
- 358 ],
- 359 "layer_b3": [
- 360 "cc6d383576c35a9709798d2e2b9e3eb31ba8c608040cf3712bc37871cfd14e21",
- 361 "ddd54c44fa28fbddecf7cfae91cfa04917fd2f2fa94fc78c528cef2356a4ec3a", # sai
- 362 "90c694e7d1e20e6da49b571e9954338d384775419790be315304103227b1051b",
- 363 "9e85aec1bdb616f52f88c80ddc7ab1eae8c16c0b5fbfcdb61a71ac02c325003d",
- 364 ],
- 365 "layer_256": [
- 366 "3c13e6a965d03a49227d8b1606ba6a343a23772d8768407cc78d4ddb9102bc80",
- 367 "b356cc84a23bc93bda4cc0fce1d0ba1b8e3d5a521e659ffc72e9e4a2d2c7f204",
- 368 "270df7317fe01abf06333acbbd4f15f8fc7a7c56053219f42efb598454a3af24",
- 369 "7ab6aa4514dd09f3cf589587d51a81734193ce45dd51bda9db0bd62fe48ef7d5",
- 370 ],
- 371 },
- 372 ),
- 373 (
- 374 "Tencent-Hunyuan/HunyuanDiT-v1.2-Diffusers",
- 375 "HunyuanDiTPipeline",
- 376 {
- 377 "identifiers": ["extra_embedder", "model.blocks", "skip_norm.weight"],
- 378 "file_256": [
- 379 "4fb84f84079cda457d171b3c6b15d1be95b5a3e5d9825703951a99ddf92d1787", # normal
- 380 "e01db5e129e8ca1117e9cf473fc5a2b096949f03ab90048aeabbc328de7ec800", # distilled
- 381 "8af691cadb78047d55721259355d708e87ddbba1b7845df9377d9a5ae917b45d", # 1.2
- 382 ],
- 383 "layer_b3": [
- 384 "aead6b61b17ebc77c4c186a4b82c193f11ec267b20d909726422ee9852e2e0b2",
- 385 "885a056b94f6f9844c0660be489844d63bb74cc13316f441d10968fff3dd3120", # distilled
- 386 "390d951cbdda6e2cffb690031b60f02921624651534c2effaaa7d68ab476c700",
- 387 ],
- 388 "layer_256": [
- 389 "d4842ce2b7f927203326b25ff4d6738ec9a8b95327f06791c387e4a351ed6ed0",
- 390 "5af943f96f5dc9fecb1e92fe2b1fa17c94dd6947690201f4a5ee1a4a2721a68e", # distilled
- 391 "4a1f2b8234fa4336e263842e042d42e8d64d8a4d3941d9c0c78366b50303950c", # 1.2
- 392 ],
- 393 },
- 394 ),
- 395 (
- 396 "Alpha-VLLM/Lumina-Next-SFT-diffusers",
- 397 "LuminaPipeline",
- 398 {
- 399 "pkg": {
- 400 0: {
- 401 "precision": " ops.precision.bfloat.B16",
- 402 },
- 403 },
- 404 "identifiers": ["time_caption", "feed_forward"],
- 405 "file_256": [
- 406 "371153b7c7b7a64899d4016970c7cc472039f9c9b21ebe073adf0b8525cdf1bd",
- 407 ],
- 408 "layer_b3": [
- 409 "fa134efd6e9672e7de2965e4895fc58879bd0a6c4fdf9165c278f2748254675f",
- 410 "4d960ec35c53f72f065b94b836bcd923ea6074d38ad49881061f315d62e3c839",
- 411 ],
- 412 "layer_256": [
- 413 "3938a85568d9df186923edf04391d79e89e6199123bc175afb520e0948d1ae05",
- 414 "c0ca51fdea051fcd042bf4b56d32e1e8bb9525a921f2e197f370f101e90527f0",
- 415 ],
- 416 },
- 417 ),
- 418 (
- 419 "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS",
- 420 "PixArtSigmaPipeline",
- 421 {
- 422 "identifiers": ["adaln_single", "scale_shift_table"],
- 423 "file_256": [
- 424 "c34b520ef473329b945c2a21083cdf1337c5a468d23b3215b65576789bfd0305",
- 425 "2fa4dee9229c02b03163f57bdb8e80c7a5ee364b7161796abe9c05e8dd13f239",
- 426 ],
- 427 "layer_b3": [
- 428 "a199930ff537994872da77391955f0dd52eddd22ab9105388f0c5852f1b8021f",
- 429 "ee6f980c32e98da6885f3e97d3f88d9158031e362cd3a49b20d1e23924b251e3",
- 430 ],
- 431 "layer_256": [
- 432 "e0afd203aff5a1d192e325d0f59361373273d85d138b51768c3f10a75c154dc0",
- 433 "987f3c2ff5d399191e5fd7dd7b1f1f285c197dc8124ad77f05cde7f2fb677a3c",
- 434 ],
- 435 },
- 436 ),
- 437 (
- 438 "PixArt-alpha/PixArt-XL-2-1024-MS",
- 439 "PixArtAlphaPipeline",
- 440 {
- 441 "identifiers": ["aspect_ratio", "y_embedding", "emb.resolution", "caption_projection"],
- 442 "file_256": ["809a92d52a4a228f381a4b4f4b76051294b73285fb0cbb02f0ad24f9372217a8"],
- 443 "layer_b3": ["c5be83545ce9dbc564bcc9fd8fe4157d131347ccfc8f62adc877ec205b20acee"],
- 444 "layer_256": ["117225c0e91423746114b23d3e409708ad55c90ff52b21fa7a1c5105d2e935a5"],
- 445 },
- 446 ),
- 447 (
- 448 "stabilityai/stable-diffusion-3.5-medium",
- 449 "StableDiffusion3Pipeline",
- 450 {
- 451 "pkg": {
- 452 0: {"precision": "ops.precision.float.F16"},
- 453 },
- 454 "identifiers": [
- 455 "model.diffusion_model.joint_blocks.",
- 456 "transformer_blocks.21.norm1_context.linear.weight",
- 457 "transformer_blocks.31.norm1_context.linear.weight",
- 458 "blocks.11.ff.net.2.weight",
- 459 ],
- 460 "file_256": [
- 461 "ffef7a279d9134626e6ce0d494fba84fc1c7e720b3c7df2d19a09dc3796d8f93", # large
- 462 "11fe06e22364b823dfeedc275912336b932b32a293a0b2f35ffac071990cc4de", # medium
- 463 ],
- 464 "layer_b3": [
- 465 "e411016545785046810b29cc3999f40bc6392be134a1318386c6f1c48f98726a",
- 466 "a81e07ee67bc627e8b3c5e292ec1ca239009517a2106e8249d670ced0a88f746", # med
- 467 ],
- 468 "layer_256": [
- 469 "13c982a6dc82d21c9f459e837d8c6f6d4696fd6e7e7b5783bdd2250b1f4fec61",
- 470 "6ee79050373337bf63ac20916596df778bb22022bb38af986128a7459eda1463", # med
- 471 ],
- 472 },
- 473 ),
- 474 (
- 475 "Efficient-Large-Model/Sana-1600M-1024px-BF16-diffusers",
- 476 "SanaPipeline",
- 477 {
- 478 "pkg": {
- 479 0: {
- 480 "generation": {
- 481 "height": 1024,
- 482 "width": 1024,
- 483 "guidance_scale": 4.5,
- 484 "num_inference_steps": 20,
- 485 },
- 486 "precision": "ops.precision.bfloat.B16",
- 487 },
- 488 },
- 489 "file_256": [
- 490 "b0b50c33be8758713459aa3c760feef6315d4bea31521fb5b8c3e8fdd9841ffe",
- 491 ],
- 492 "layer_b3": [
- 493 "461e3d83dfa7e075ef21e2138ef153922ecfadde3db464b03dff92819f3e86dd",
- 494 ],
- 495 "layer_256": [
- 496 "b928bbcc2ce99d55d21c189e2b1c57498bc313ef5b1457036e356107d567fc4e",
- 497 ],
- 498 },
- 499 ),
- 500 (
- 501 "stable-diffusion-v1-5/stable-diffusion-v1-5",
- 502 "StableDiffusionPipeline",
- 503 {
- 504 "identifiers": ["up_blocks.3.attentions.0.transformer_blocks.0.norm3.weight"],
- 505 "file_256": [
- 506 "6ce0161689b3853acaa03779ec93eafe75a02f4ced659bee03f50797806fa2fa", # pruned ema only original safetensors
- 507 "1a189f0be69d6106a48548e7626207dddd7042a418dbf372cefd05e0cdba61b6", # pruned original safetensors
- 508 "e1441589a6f3c5a53f5f54d0975a18a7feb7cdf0b0dee276dfc3331ae376a053", # ema pruned original ckpt
- 509 "cc6cb27103417325ff94f52b7a5d2dde45a7515b25c255d8e396c90014281516", # pruned ema original ckpt
- 510 "19da7aaa4b880e59d56843f1fcb4dd9b599c28a1d9d9af7c1143057c8ffae9f1", # diffusers safetensors
- 511 "cd1b6db09a81cb1d39fbd245a89c1e3db9da9fe8eba5e8f9098ea6c4994221d3", # diffusers non ema safetensors
- 512 "c83908253f9a64d08c25fc90874c9c8aef9a329ce1ca5fb909d73b0c83d1ea21", # diffusers fp16
- 513 ],
- 514 "layer_b3": [
- 515 "909c6ff3192ab2767e789a6125865bc23163db467ab78b1c633bad46a4293fad",
- 516 "b52807536902cabbf84f99e4fa2f8713fb4ef77e739f06367ee0d486e3222faa", # ckpt
- 517 "d31382d71a1044b636d80d861a2b4dbca51826bed34d34b5c14608b7679ccefd", # safetensors ema pruned
- 518 "5fd8b28013b7e5a64c7c235f0a93d93e48bc19a0e5dde7b646a87b429219643a", # safetensors pruned
- 519 "731f552f29edcb4f86112cc94d296377f3533a9633ccf83e202d9e1785d94a00", # diffusers
- 520 "2d2f97574a161cf01a6f6d476b141c7be06f940d94b695ffc12c4e74eca2de1c", # diffusers fp16
- 521 ],
- 522 "layer_256": [
- 523 "ece771354ad470a82d56eda413ae3dd6c00d2de28ab3c56a88201d08d4424b4b",
- 524 "65b084dada803461ab9ca9be9b892d211870a121dd6c555a111eea470b951c54", # st
- 525 "dc937b59892604f5a86ac96936cd7ff09e25f18ae6b758e8014a24c7fa039e91", # ckpt
- 526 "92565dec90f7c8412dc872e820f66cd0c56263bbbc392439645b6fee270f41bb", # st fp16
- 527 ],
- 528 },
- 529 ),
- 530 (
- 531 "Kwai-Kolors/Kolors-diffusers",
- 532 "KolorsPipeline",
- 533 {
- 534 "pkg": {
- 535 0: {
- 536 "precision": "ops.precision.float.F16",
- 537 "generation": {
- 538 "negative_prompt": "",
- 539 "guidance_scale": 5.0,
- 540 "num_inference_steps": 50,
- 541 "width": 1024,
- 542 "height": 1024,
- 543 },
- 544 },
- 545 1: {"diffusers": "DiffusionPipeline"},
- 546 },
- 547 "file_256": [
- 548 "425ff1dcbe3a70ac13d3afdd69bd4e3176b0c3260722527c80b210f11d2d966c", # fp16,
- 549 ],
- 550 "layer_b3": [
- 551 "6eb15506fa38b4cbb26391ab1b6c9ead05f86c711e46583bfbe8fc4421571414", # fp16
- 552 ],
- 553 "layer_256": [
- 554 "04e3c17170b8a200481f6941b370fdc5056a00fe5a16956de01790f8a93c0dcd", # fp16
- 555 ],
- 556 "identifiers": [".DenseReluDense.wi.weight", "encoder_hid_proj.weight"],
- 557 },
- 558 ),
- 559 (
- 560 "stabilityai/stable-cascade-prior",
- 561 "StableCascadePriorPipeline",
- 562 {
- 563 "pkg": {
- 564 0: {
- 565 "precision": "ops.precision.bfloat.B16",
- 566 "generation": {
- 567 "negative_prompt": "",
- 568 "num_images_per_prompt": 1,
- 569 "num_inference_steps": 20,
- 570 "guidance_scale": 4.0,
- 571 "width": 1024,
- 572 "height": 1024,
- 573 },
- 574 }
- 575 },
- 576 "file_256": [
- 577 "673b3173b037fb5f65b14fde37267390641a36726683de75dcf9df76fce2b866", # lite bf16
- 578 "45c1eb5ce9b69efac891ad459b15c215cd90a986adbbfaf3effd3a89578cbcaf", # pretrained
- 579 "088ddf1e444abf399007b2da2bac87791df165c69f477994f6b3c745a20904b0", # stage c modelspec sai
- 580 "39cec96c7212607f9e526db719bf1df507166d09f4748676c13b0d31cd4adb07", # stage c
- 581 "31ffe2f1a3e2351d658fc7d3002a4eca22466a680f7fb3715b1e3768476f9633", # stage c lite
- 582 "dfe24009fc881011f350d08d9d13be13a1a3b3cbfed667435efe0fd419aca099", # bf16
- 583 ],
- 584 "layer_b3": [
- 585 "c55c83fa435ed128457f605bf1312e54727996d1c94413fc5ab5b49e9933857c",
- 586 "6fb07ed9fc6ee636e50783802754b3a37bbecfc67037813b616223aeaf6fe877",
- 587 "2ea194240e105c8962923e2baca88cb6a0c826794afc2ef82474301694711d68",
- 588 "3412c8a184805621e4595d57268ced0b5c3c1974cd221bf67b2c908eec4fd61c",
- 589 "53abfb013cfb0e41d0bc7b96bb83e42a4d4c67cb7325f9acf645b02d90efd8fe",
- 590 "34556558f680c183adc2accd493cb9888a98ba853226bbecb07d95eb2055ff4f",
- 591 ],
- 592 "layer_256": [
- 593 "4f5e0a738b963d3d4f8413387a0966ac1ce51f0f985bcbcc124fa221a2fff467",
- 594 "8aa77e732a398b7d0dcd9a35d5682c2b5ab090ae90e915c7c91878abff0284d8",
- 595 "4bbd46ded0916de3108f0da7145a80f5c7acea26ed35b0aaa29af12008352453",
- 596 "415d1f3ecd06416708c1b83ab21e50b39c9d88d19dc33e60b977b7b7061880b9",
- 597 "f678c32815c238e14091f690c8a83c3375c8f7738dc7abff79ff086ed9b59204",
- 598 "17c8da803df7b9bbc8b1d7cc0c44916fea5b5ac0891330c4fdf0326fcd4496cb",
- 599 ],
- 600 "identifiers": ["down_blocks.0.2.kv_mapper", "previewer", "backbone"],
- 601 },
- 602 ),
- 603 (
- 604 "black-forest-labs/FLUX.1-dev",
- 605 "FluxPipeline",
- 606 {
- 607 "pkg": {
- 608 0: {
- 609 "precision": "ops.precision.bfloat.B16",
- 610 "generation": {
- 611 "height": 1024,
- 612 "width": 1024,
- 613 "guidance_scale": 3.5,
- 614 "num_inference_steps": 50,
- 615 "max_sequence_length": 512,
- 616 },
- 617 },
- 618 1: {
- 619 "mflux": "flux.flux.Flux1",
- 620 "generation": {
- 621 "height": 1024,
- 622 "width": 1024,
- 623 "gudance": 3.5,
- 624 "num_inference_steps": 25,
- 625 },
- 626 },
- 627 },
- 628 "file_256": [
- 629 "f6315581b7cddd450b9aba72b4e9ccf8b6580dc1a6b9538aff43ee26a1a3b6c2", # krea sai
- 630 "1b2170ac37156d4cf91909eb6834bb8adac84bc1fce8098a29cfb03738df84ad", # krea diffusers
- 631 "4610115bb0c89560703c892c59ac2742fa821e60ef5871b33493ba544683abd7", # modelspec sai
- 632 "d86a3038eacaa720682cb9b1da3c49fecf8a3ded605af4def6061eaa18903eb8", # diffusers
- 633 "b7d840eef01c27dfd72ae9143c261355a51bab3b2662263a6cb0059d55347c3d", # qwen2
- 634 ],
- 635 "layer_b3": [
- 636 "261559c8eaccae558f72621804a9ee188d338e45e2c622a58db709ac190198ba",
- 637 "87f5d565c66e40eb02eb96498243ad81afcbf86192db99a4fc8fff215470320e", # modelspec sai
- 638 "e61d10a394902dadca9367467b2245070f651f4553ec4a96192fbba64e820acb", # diffusers
- 639 ],
- 640 "layer_256": [
- 641 "3db58cf834d2f81abb1e035131956da4c90451074c681d0db10810e55e60c2c4",
- 642 "ddf1a34a06b355ce2bcd0f9beb0713450d9bcdc61a03a6bc37716361735e96f1", # diffusers
- 643 "ad8763121f98e28bc4a3d5a8b494c1e8f385f14abe92fc0ca5e4ab3191f3a881", # modelspec sai
- 644 ],
- 645 "identifiers": [
- 646 "double_blocks.12.txt_mod.lin.weight",
- 647 "add_q_proj.weight",
- 648 "single_transformer_blocks.9.norm.linear.weight",
- 649 ],
- 650 },
- 651 ),
- 652 (
- 653 "black-forest-labs/FLUX.1-schnell",
- 654 "FluxPipeline",
- 655 {
- 656 "pkg": {
- 657 0: {
- 658 "precision": "ops.precision.bfloat.B16",
- 659 "generation": {
- 660 "height": 1024,
- 661 "width": 1024,
- 662 "guidance_scale": 0.0,
- 663 "num_inference_steps": 4,
- 664 "max_sequence_length": 256,
- 665 },
- 666 },
- 667 1: {
- 668 "mflux": "flux.flux.Flux1",
- 669 "generation": {
- 670 "height": 1024,
- 671 "width": 1024,
- 672 "num_inference_steps": 4,
- 673 },
- 674 },
- 675 },
- 676 "identifiers": [
- 677 "double_blocks.12.txt_mod.lin.weight",
- 678 "add_q_proj.weight",
- 679 "single_transformer_blocks.9.norm.linear.weight",
- 680 ],
- 681 "file_256": [
- 682 "9403429e0052277ac2a87ad800adece5481eecefd9ed334e1f348723621d2a0a", # sai modelspec
- 683 "9b633dbe87316385c5b1c262bd4b5a01e3d955170661d63dcec8a01e89c0d820", # diffusers
- 684 ],
- 685 "layer_b3": [
- 686 "c65ba812ce3ce056eb1585673f62fb896afe6ec049faaf00a97bc35c9a398c44",
- 687 "03049273329fc7db2da10de6d3eb27cb03f190e379c0556cc97b3f0f29001d0c", # sai modelspec
- 688 "483c4be8ef031c56bc8450d1a3cfbe54445ed317bcd801be5abe89f1d3c48790", # diffusers
- 689 ],
- 690 "layer_256": [
- 691 "79c07e339865fe9e22c80f723d728c778130acd07a330339c68218b92bb7b3b8",
- 692 "ef5c9cd1ebe6e3be5e8b1347eca0a6f0b138986c71220a7f1c2c14f29d01beed", # sai modelspec
- 693 "27bc71eca2d2ff7459165acc12010230911db7709a4f6a5c255befedfa6b1649", # diffusers
- 694 ],
- 695 },
- 696 ),
- 697 (
- 698 "stabilityai/stable-cascade",
- 699 "StableCascadeDecoderPipeline",
- 700 {
- 701 "pkg": { # prior=prior_unet
- 702 0: {
- 703 "generation": { # image_embeddings=prior_output.image_embeddings,
- 704 "negative_prompt": "",
- 705 "guidance_scale": 0.0,
- 706 "output_type": "pil",
- 707 "num_inference_steps": 10,
- 708 },
- 709 "precision": "ops.precision.bfloat.B16",
- 710 },
- 711 },
- 712 "file_256": [
- 713 "fe92687deefcfb33bb3ec181254b55fe4e434c5084ce9d38815eaa32487ad376", # lite bf16
- 714 "2c8d58b267678aecfa6705a0a0375c88613065a8a8d32ad3a4c3867f5461cb3a", # bf16
- 715 "6c218dc948575e3b14b03dffe2014d7870ac505005770ce3abdc28e920a03c05", # b modelspec sai
- 716 "a6c3d534a9be308e95d2c3224af94a854bebd9b503f620f1ae3c8e6ba4a341bf", # lite
- 717 "7b431ea7d0f10e72b3eaece353bf6bf2f6bc717b6f4207411be186b40dec1f43", # b
- 718 ],
- 719 "layer_b3": [
- 720 "9506d989de0226018de214f7ced4670eb5aad4a0c399a9229488ceccdf9a3ceb",
- 721 "6c09dcb83e0cd7ad735eb763c5e3721c579d796853f0b9d31ba74fb13cad4f94",
- 722 "e07025965cee925e31f1d617ea8baa575e7db910d40cc0482fd83df317c0812b",
- 723 "d9a42e4226fb2778aaeaf0d6bda173a4ff95aa574c6d9e27e41542aa469e40a3",
- 724 "8dcd87dc7a9b877e8e2a00abac44c4da9eadf2b8df4ae68f27415bb791381a96",
- 725 ],
- 726 "layer_256": [
- 727 "630ec0f3adf97145316c034139836f9df952060d0237ac4e478c55d9a3a50bc8",
- 728 "80904f707c192ddd06be2cebeb2ebbec3eb0e9c99076d50824d391ef3ac67bf2",
- 729 "8ccedbe1e8cc4093f05b5f8d90e6103e688ae1ac71e0d6261fb17c42ff7c25e4",
- 730 "3524e7fa9ca6f7ef695bc2d3410934eabd5272946a05c8cacd7f329e0bd9f1dd",
- 731 "40499a8f45ae28558ed2fe4fc549a4cb469bd237434b331ccc0b1910310ed733",
- 732 ],
- 733 "identifiers": ["0.2.channelwise", "clip_mapper.bias", ".12.self_attn.k_proj.weight"],
- 734 },
- 735 ),
- 736 (
- 737 "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers",
- 738 "WanImageToVideoPipeline",
- 739 {
- 740 "file_256": [
- 741 "b4602c35fa0519750a42c03e3f296c02d542291e344c4d702522cddbd1711f13", # 480 diffusers
- 742 "6d7a34b63b70eb608324e546d979167a5e787ac6bca3528e63f54a11572d66aa", # 720 fp8 scaled sai
- 743 "b2051cd29d6b2f0c924fa7a3e78a4772f0134d7b059f21590dcce416f4f6cbe8", # 720 fp8 sai
- 744 "7664fe075b3c82dcecf89012ad3429eee41ee9f10d476f60bc2d2ae3c4ca986c", # 720 fp16 sai
- 745 "8ef7ea5bf9eea636b9b3ebd84c40671b4a18ae2704cb4c8595cb5b25c1d8e8b9", # 720 bf16 sai
- 746 "b2de21b99b2e72cb0ff15253b07e926f26e7cf1b7e229efc32f94ad1f1ed9395", # 480 fp8e4m scaled sai
- 747 "0ca75338e7a47ca7cacddb7e626647e65829c497387f718ecb6ea0bae456944a", # 480 fp8 scaled
- 748 "c058a4ac5363c35d1ab4dd3bdec788c23b267fa42a0d7c68aba599f2f74600c9", # 480 bf16 sai
- 749 "27988f6b510eb8d5fdd7485671b54897f8683f2bba7a772c5671be21d3491253", # 480 fp16 sai
- 750 ],
- 751 "layer_b3": [
- 752 "4b6c3354c9ee5694e00a78f5658fdf14129f159c3b78a57f82fb18e0f265a83d",
- 753 "c36c783559a40d22504f6c4bfb4f5aae760f3f46bbb3a595be79880935122175", # fp8 scaled
- 754 "ac62f7d5583fd2e85b738fafaf233e2cde6e2857e04351135bb9ded45f9082ce", # fp8
- 755 "215e89e855b5e9456af9aa68bc67567dc2269002aaa6b01d849ffec425fc628d", # fp16
- 756 "324b8b6c2d512547a2c31bafa12e20acf313fd3aad587b293334f9f629edeec6", # bf16
- 757 ],
- 758 "layer_256": [
- 759 "137881dad8c00063bc8bf05f93067736e419173cd171acc22f77b730db688a19",
- 760 "8c5952fd3d333d3a4b719bf7d8ce6b12d1d2e78caaa7e42d713788cfdcadd244", # fp8 scaled
- 761 "86c58bc4864c97f394ea6bccb2ecedc4aab7166f5b9bfeb313edfdcb2918164a",
- 762 "cac45f7d8f1a0628cb0738bd308689e439b1cc6206e5f887d60d5b37d30138f2",
- 763 "60e4f71a0961b1346b6f6b5ebe4c8cc93219239c5e13b4c0f1e19e9b8e1324d5",
- 764 ],
- 765 },
- 766 ),
- 767 (
- 768 "Qwen/Qwen-Image",
- 769 "QwenImagePipeline",
- 770 {
- 771 "file_256": [
- 772 "9f33a59093af3abcc2836d4cf4b7bd122c238ca70a26c70f34fdde64646b3bcd",
- 773 ],
- 774 "layer_b3": [
- 775 "c87eedda853c12844a8deb3592a90bbcbd4dff2f7a850c28755e4aa171432150", # diffusers
- 776 ],
- 777 "layer_256": [
- 778 "fda2472d8ef6587a4c979021a2390eeb7c8fc2bcf565330ab8dc6b22f5348ec9", # diffusers
- 779 ],
- 780 },
- 781 ),
- 782 (
- 783 "Wan-AI/Wan2.1-VACE-1.3B-diffusers",
- 784 "WanVACEPipeline",
- 785 {
- 786 "file_256": [
- 787 "bd8bbb8834a274525ab65cbb063f21aa58973a054bfd1638bfe395504c9d9b99", # diffusers 14
- 788 "192804a4e10b5bb0a13f5c224bc4ec9707b3b8cc0def8eea005dbce7c9d6752a", # diffusers 1.3
- 789 "f202a5c59b8a91ada1862c46a038214f1f7f216c61ec8350d25f69b919da4307", # 14 fp16 sai
- 790 "654693bf2a93a27cd67c3bcee238bc1d0cbb0dd9a74928ed7155fb21a2a1900a", # 1.3 preview fp16 sai
- 791 "640ccc0577e6a5d4bb15cd91b11b699ef914fc55f126c5a1c544e152130784f2", # 1.3 fp16 sai
- 792 ],
- 793 "layer_b3": [
- 794 "5357d78799a61cd2d72a8a2824c919d63f718eb3fba624af63689e9c657db032", # diffusers 14
- 795 "7ae67b7ccf79d1c3f4531ae138e1eb63d52dd97a66b3fcbe1d68fded8df4d5b1", # diffusers 1.3
- 796 "ee63ecdfb3da6901853a59ec950f3e7c3f6595ac46347a03881a4a9c71425377", # 14 fp16 sai
- 797 "82762df3539021d3c0342e0da04137ddbe95ef37ea933cd0a68c09c2c650f2ac", # 1.3 fp16 sai
- 798 ],
- 799 "layer_256": [
- 800 "2684413479030170fb3f08c1069c02957ffc386a59168d23b55d579d5c675269", # diffusers 14
- 801 "d527680fa735e5f30ef8852aabf8a49f02a094bc4718f0787c5b85710a13c026", # diffusers 1.3
- 802 "9677492a107b3ed827c7285db3393f5321d451cc6d922a4d0488d2a67e939446", # 14 fp16 sai
- 803 "aaef66a4f65ecf852888d160b2122753fe4c6d642b5d41db29e4ce9e6855b5a0", # 1.3 fp16 sai
- 804 ],
- 805 },
- 806 ),
- 807 (
- 808 "Wan-AI/Wan2.1-T2V-14B-Diffusers",
- 809 "WanPipeline",
- 810 {
- 811 "pkg": {
- 812 0: {
- 813 "precision": "ops.precision.bfloat.B16",
- 814 "generation": {
- 815 "height": 480,
- 816 "width": 832,
- 817 "num_frames": 81,
- 818 "guidance_scale": 5.0,
- 819 },
- 820 },
- 821 },
- 822 "file_256": [
- 823 "299e6304544f2783896372fa919e755a8bb9ab8caf898ce08a678dae391e1179", # diffusers
- 824 "a9278e6e9c82d174e6c67b3c97d8b97fef30af51dcf59160f2fc241f6819f5dc", # diffusers 2
- 825 "be531024cd9018cb5b48c40cfbb6a6191645b1c792eb8bf4f8c1c6e10f924dc5", # fp16 sai 1.3
- 826 "6f999b0d6cb9a72b3d98ac386ed96f57f8cecae13994a69232514ea4974ad5fd", # bf16 sai 1.3
- 827 "2e39adde59c5e0e90edbb35873126b0d67928b5c11c501e384e976d6dc597cce", # fp8 scaled sai
- 828 "2ee88ab18d7ed7691c5b7f8bdc3d0a9815e6efe75499287564830fd209d3cdfb", # fp8 sai
- 829 "46c27d3693bf2475990a912e08bf67fc6e6cd5396eab87b5e8dd1fcd3651364a", # fp16 sai
- 830 "193535c6450045f718df5f011de6d94d49bd9b13f37ca0412500f050dbbb01a8", # bf16 sai
- 831 ],
- 832 "layer_b3": [
- 833 "32266d1c79b518adb9d21837e6a427f6ae55b68cfdd673a7dadb38820fddeb48", # diff
- 834 "3b6989856f4f05368524c1852d8660b73c84cfbe44460af017d7139c2a4641b8", # fp16 sai 1.3
- 835 "f4d6cee3c112db93b3c9137ad102ec0e79ec7ab68b9bbc59004fbc268ccd5ddb", # bf16 sai
- 836 "e627144f41055619eb5407699c46e69ac0d87cf8873721e3e48c9e842656abf8", # fp8 scaled sai
- 837 "6c00f3fadedacb841c4b9b4321b94a11ef85a08c9dd9253e5f9ba95856715579", # fp8 sai
- 838 "a0c339253c714b05877c8fbab649ed631cf021930978f3696a46f685a07c9092", # fp16 sai
- 839 "6435da89a870fd0e88680d31de75b9a40c408a4768eff384ce9b9e99481e8e66",
- 840 ],
- 841 "layer_256": [
- 842 "52493c23c5fc1d087a283bc4eabb151421b7ae09affa12a5bb059d62656c5766",
- 843 "058dedb3d2683a9a5b671c6302690e22722c93f6ed92281d5fa74ab190e632a1",
- 844 "5fbed4b95e7196d3626003ea9e0fbbffd074b4297ca406e01b5b6c5d881a6080",
- 845 "3a2335c8e7a4359c071b50333b5c00eef6f42a1d5206915e2ee99464a8c5eae7",
- 846 "0542780670dd75d4cd9deda123d2e150730646c0a1a8d34582460991498a77a6",
- 847 "e925b8222774905c8fbf10af77811fde7870e563eedcde2c94bd5c727e952d49",
- 848 "3d915854976284347efa7aa0a117c0fc3b415c4208e1a6c94beb4ccb9720743d",
- 849 ],
- 850 },
- 851 ),
- 852 (
- 853 "Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
- 854 "WanVideoToVideoPipeline",
- 855 {
- 856 "pkg": {
- 857 0: {
- 858 "diffusers": "WanPipeline",
- 859 "precision": "ops.precision.bfloat.B16",
- 860 "generation": {"height": 480, "width": 832, "num_frames": 81, "guidance_scale": 5.0},
- 861 },
- 862 }
- 863 },
- 864 ),
- 865 (
- 866 "nvidia/cosmos-predict2-text2image",
- 867 "Cosmos2TextToImagePipeline",
- 868 {
- 869 "file_256": [
- 870 "7fbd20dae97cc26a55c7aff3024bc84e554cff8f69966c725a24c8238c5431ec", # gguf
- 871 "6d211f1c14cd793156da3a840dd5462ae072046fcd6f1dc64c613a5343bfe896",
- 872 "95a2b32ad31a271eb64d35985c7ea46f1448528af70932eb1f35d57f90c27be2",
- 873 "344e67faf333b7849fa94290c9028bdd5e40eb19700754c833cda0423bc10ad0",
- 874 "ce15ef565cbb9ef414a6f7a396c455d82d5f762d2174493da87fe009c5fee75b",
- 875 "94aa9f2b59330b88e97b6b439e2f206a51c86e6b154fb66d43ed149bfac23cf8",
- 876 "636de5388da249130d51752991a1792b90af31cbf43f021ae07f75756ee2d79a",
- 877 "472c5e4cf5056a1a59085addb5a86d801de39bf5e000d253f206a7f63c710029",
- 878 "663266ace67c22529c3b6bfa0e8bd69f0ba6e683f5f02b8e3da50881057ba142",
- 879 "21a674b314c1364d0dbb3712f5ed702996a7b7403c452835cac22709e01c2f77",
- 880 "3bf2df806c6472e039efc9e8d3181163d7faa7b385e61519b7d17d5e9c993a49",
- 881 "1de35e1603c4c30bc80b132ccea15fc0503369caf68290708f17e679e98cd41f",
- 882 "0738e559bbd71f7351ccba34b2b47362a3f829b92f3dbcffeaf1e44b0d52f42c",
- 883 ],
- 884 "layer_b3": [
- 885 "5a18ba14c41c6601dcc1195ca180ac7744357eb15ace39272788bda1a7151e9b", # gguf
- 886 "67cc3eaf7987c89cd7ccff13de6bc03e3eec59d260d44486e2367cd946ce6f20",
- 887 "3c6fefa107742488d2e6856714198a762f2fd35c67edd50d4657eaf4b59c7ca3",
- 888 "4e1f90ee1e8959d334c9b1ea2cc5e58d0b8340e271c35f81c8a5ec26e16d9d76",
- 889 "f8171071e828524fcc2806126ad100a2198e450c82c0864c8fe8b358c5cbbfbd",
- 890 "8126101a0207ecfbd741394fd59f306bcb4c492b2a921e0921c426ca7bd38985",
- 891 "c942c5a85ff7cb602d8ca894f5d180c2224e91f0b62c3a21f6a425f9e0e8554b",
- 892 "c8c500de74da879a547875fe1046f62ab18bdfd09c09eb3da723cbc2319cb4e3",
- 893 "c0ac3f67501004e9e9a55d1658402ad97e42bf8a266edf81f6f3bb835ee476b9",
- 894 "84f5926eb4e11d826815682b076ed7d3bba4c86520859be80aa1ef92c72b26a4",
- 895 "1d4375aab5548708559b0fde150754a2163cd211eb20a5471e17afaeeb26e082",
- 896 "68bd8982f59c60d69c301d16dfb5a60f5d43d66c0b60138d48a22f5ded598e7b",
- 897 "c3e9a10cad7aebf979072092008be6e2815d03d28cbf316c15e8daf22116bd7d",
- 898 ],
- 899 "layer_256": [
- 900 "38f2a75eab667c0cc85f3946a23ca6dc2278438c25a9f93aaaa9f79c3808e180", # gguf
- 901 "ee8434a5e9bc6fa07199de2d0c69fb87f7922c31792bafd13f527c9d92fecb0c",
- 902 "2f8382657babb4d0ae4f8e425ae33b21ad71deb6ba457fd6734f05208d52e06a",
- 903 "34b181a8291b571857cdbf67ac0081fea594a2f223bf20bd2fc8b0c889e9602d",
- 904 "d198c412b972e381acfb812304fa98ed0d97a2f072ddc195cd9a1eb83b1d8146",
- 905 "79580a13aff9859e67b0a9f4f8893236cdcfa58c3d43770641aaac8daee55a94",
- 906 "cfd48c7ad71c913fa8768167ed0c2ee8c207311b22b1e5a8761369b5a780e8d6",
- 907 "da91362ad85d4d2e80a2cb7a55e4ae0e52c9eef8b437a95894ce5ab75d36568c",
- 908 "15f84001f5205b6dd8c6f1334cb51c46f6171c7795fb2a557ea16b874f0c71e5",
- 909 "5d29179ad15a15d2561defcdda66f1d1e4d065c1e0738f9cba4db5b68b93d2ea",
- 910 "7ec489d1e461f5fb2af627b68034ca57f19c516aeccbc5d188b3bd27e3353a15",
- 911 "c8dc42fe7b411d746ebdf86286b91cd6893c5f028076b8fe4103f7ea8e1d8833",
- 912 "86df7c095aee01588e961438f322b85ca0100a9e440b8a2b6c724e00f748d8b5",
- 913 ],
- 914 },
- 915 ),
- 916 (
- 917 "rhymes-ai/Allegro",
- 918 "AllegroPipeline",
- 919 {
- 920 "pkg": {
- 921 0: {
- 922 "precision": "ops.precision.bfloat.B16",
- 923 "generation": {
- 924 "guidance_scale": 7.5,
- 925 "max_sequence_length": 512,
- 926 "num_inference_steps": 100,
- 927 },
- 928 },
- 929 },
- 930 "file_256": ["6927dcc812841c1da549bf11c97ddf30532aee0e708a6642fa64cf8e0dfcdef7"],
- 931 "layer_b3": ["8b20714a6af89ea4bf4ada1f805c5b9d529ef136c229e9b75392242d62d80c3e"],
- 932 "layer_256": ["9e44e6c919dc71c24a193641e6265cd9983a2a773b9bbaf527c10ac4837b29fd"],
- 933 },
- 934 ),
- 935 (
- 936 "audioldm-s-v2",
- 937 "AudioLDMPipeline",
- 938 {
- 939 "file_256": ["fc30d5b5a3bb8d08672736efb1fff10755ba7024dace39b2dcb579a105aa2a5a"],
- 940 "layer_b3": ["82fbcc553c1ad770d28fd1866b935249c5ebfbf75f3166ae823e1bc6ef39a95a"],
- 941 "layer_256": ["d076446a58a36bf436e37444679d62bcf2f45689d4aa3d799b3fe801c71ed2c8"],
- 942 },
- 943 ),
- 944 (
- 945 "zai-org/CogVideoX-2b",
- 946 "CogVideoXPipeline",
- 947 {
- 948 "pkg": {
- 949 0: {
- 950 "precision": "ops.precision.float.F16",
- 951 "generation": {"num_videos_per_prompt": 1, "num_inference_steps": 50, "num_frames": 49, "guidance_scale": 6},
- 952 }
- 953 },
- 954 "file_256": ["8fbb6a5e67c70885a8ed8e33df144ac61253e45977be5035fa18cfdf77d386c7"],
- 955 "layer_b3": ["1db3439649b5362448455fb2ed6ebde0c3b973655a206832731149757ad165bb"],
- 956 "layer_256": ["edd6bd51f1236f528ff8d32dc754f0b86cfac901b800642ea497358156dc00bd"],
- 957 },
- 958 ),
- 959 (
- 960 "HiDream-ai/HiDream-I1-Full",
- 961 "StableDiffusion3Pipeline",
- 962 {
- 963 "file_256": ["3cb3f6d77a3fce19b90fa7f66da0cbe997b0785a38a788b559290d3062f6fd26"],
- 964 "layer_b3": ["612eb9b2676a3e7b28b10aae045a97a95de2a399fe3801c8f6369589c3a832a6"],
- 965 "layer_256": ["78fbfb7fddb9ccbdf91f22b0c3d304cbf0cc7305dbccb216982233849ec727df"],
- 966 },
- 967 ),
- 968 (
- 969 "cvssp/audioldm2",
- 970 "AudioLDM2Pipeline",
- 971 {
- 972 "pkg": {
- 973 0: {
- 974 "precision": "ops.precision.float.F16",
- 975 "generation": {"num_inference_steps": 200, "audio_length_in_s": 10.0},
- 976 },
- 977 },
- 978 "file_256": ["359a5ffb89a844beb2fcfac584aae2cd7cd6e87c3ab1ec4e892ef45d91db77c2"],
- 979 "layer_b3": ["eac241273f9f30982fc04aa88b4dc1c38b533430956a55b9ed4d3e5c717ec962"],
- 980 "layer_256": ["ab109d01b43788063802f00c6ecab024c830ea58d668f5c2df9e3ae5b87d86cb"],
- 981 },
- 982 ),
- 983 (
- 984 "tencent-hunyuan/hunyuandiT-v1.2-diffusers",
- 985 "HunyuanDiTPipeline",
- 986 {
- 987 "pkg": {
- 988 0: {
- 989 "precision": "ops.precision.float.F16",
- 990 }
- 991 },
- 992 "file_256": ["7d31ac8fa389ff39dd0a81430010e52c43b59f15adc00c83625a47881e16830e"],
- 993 "layer_b3": ["bccd37ecc9f85d132b46d0bb67b4facb49fc6c091428a4feba9ab9a93140f5fe"],
- 994 "layer_256": ["ed25d241d58ca298d28abd5919e70341ad194e77dce4859436b52ea4d8fcb616"],
- 995 },
- 996 ),
- 997 (
- 998 "Alpha-VLLM/Lumina-Image-2.0",
- 999 "Lumina2Pipeline",
-1000 {
-1001 "pkg": {},
-1002 "file_256": [
-1003 "132b4d213fdd3cfc14333746fc3eb8bbe6358cd73c3bc95ac4ccec230b97dca3",
-1004 "a7c09ebae62996a8289782161338a3cdba58c11d2d849c50b2d6502e152b0d6d", # pth single file
-1005 ],
-1006 "layer_b3": [
-1007 "198bde52f09736f1fc650dcdbd0e6b0f6a5ce186582554c1d9ee8ab16ac0feb2",
-1008 "b52807536902cabbf84f99e4fa2f8713fb4ef77e739f06367ee0d486e3222faa",
-1009 ],
-1010 "layer_256": [
-1011 "982893c99860aac8198c2e435cf85f782fce8f10732daf1f2881a26864400a4e",
-1012 "dc937b59892604f5a86ac96936cd7ff09e25f18ae6b758e8014a24c7fa039e91",
-1013 ],
-1014 },
-1015 ),
-1016 (
-1017 "ucsd-reach/musicldm",
-1018 "MusicLDMPipeline",
-1019 {
-1020 "pkg": {
-1021 0: {
-1022 "generation": {
-1023 "num_inference_steps": 200,
-1024 "audio_length_in_s": 10.0,
-1025 },
-1026 }
-1027 },
-1028 "file_256": [
-1029 "853d0ef1d61cbf5d682872322ea8b761ba3d2f85bfbccd58363bd6b2f837268f", #
-1030 ],
-1031 "layer_b3": [
-1032 "82fbcc553c1ad770d28fd1866b935249c5ebfbf75f3166ae823e1bc6ef39a95a" #
-1033 ],
-1034 "layer_256": [
-1035 "d076446a58a36bf436e37444679d62bcf2f45689d4aa3d799b3fe801c71ed2c8", #
-1036 ],
-1037 },
-1038 ),
-1039 (
-1040 "openai/shap-e",
-1041 "ShapEPipeline",
-1042 {
-1043 "pkg": {
-1044 0: {
-1045 "precision": "ops.precision.float.F16",
-1046 "generation": {"num_inference_steps": 64, "size": 256, "guidance_scale": 15},
-1047 }
-1048 },
-1049 },
-1050 ),
-1051 (
-1052 "hunyuanvideo-community/HunyuanVideo",
-1053 "HunyuanVideoPipeline",
-1054 {
-1055 "file_256": [
-1056 "bdb957b35585ea74ae42ca92865a68fa1bf1ebc6c5b7e686a889e5c977dc24c7", #
-1057 ],
-1058 "layer_b3": [
-1059 "d31c56b4c9444d4c2f1b10120fe964e0956f6b8c7e7c1e4cc5a1f37406fc49f5" #
-1060 ],
-1061 "layer_256": [
-1062 "fe741fdfd163bcb1e0ed81d80f79ac3576dbf6e6740674efadfeff782a48bed4", #
-1063 ],
-1064 },
-1065 ),
-1066 (
-1067 "zai-org/CogView3-Plus-3B",
-1068 "CogView3PlusPipeline",
-1069 {
-1070 "pkg": {
-1071 0: {
-1072 "precision": "ops.precision.float.F16",
-1073 "generation": {
-1074 "guidance_scale": 7.0,
-1075 "num_images_per_prompt": 1,
-1076 "num_inference_steps": 50,
-1077 "width": 1024,
-1078 "height": 1024,
-1079 },
-1080 },
-1081 },
-1082 },
-1083 ),
-1084 (
-1085 "stabilityai/stable-audio-open-1.0",
-1086 "StableAudioPipeline",
-1087 {
-1088 "pkg": {
-1089 0: {
-1090 "precision": "ops.precision.float.F16",
-1091 "generation": {
-1092 "num_inference_steps": 200,
-1093 "audio_end_in_s": 10,
-1094 "num_waveforms_per_prompt": 3,
-1095 },
-1096 }
-1097 }
-1098 },
-1099 ),
-1100 ]
-1101
-1102 transformers_addons = [
-1103 (
-1104 "google-t5/t5-small",
-1105 "T5Model",
-1106 {
-1107 "identifiers": [
-1108 [4096],
-1109 "encoder.embed_tokens.weight",
-1110 "text_encoders.t5xxl.transformer.shared.weight",
-1111 "t5xxl",
-1112 "encoder.block.0.layer.1.DenseReluDense.wi.weight", # small\
-1113 ],
-1114 "file_256": [
-1115 "ec87bffd1923e8b2774a6d240c922a41f6143081d52cf83b8fe39e9d838c893e", # shuttle/flux diffusers# flux dev
-1116 "565cb2487351282e8e4dbeb88e63f4ad28217ce0439f5a8e6525a924807d2d9b", # bf16 modelspec sai
-1117 "6e480b09fae049a72d2a8c5fbccb8d3e92febeb233bbe9dfe7256958a9167635", # fp16 modelspec sai
-1118 "4f2751ceeb2a96edd693e539dc5d6bba0b8d3814f49a9b3798403a0cec4b2e3d", # fp16 diffusers cogvideox
-1119 "83690f3cc37cecb5e907f41ab0f7abb0855ef24a0a8aab9259f2888ce85a34e2", # flux diffusers
-1120 "7d330da4816157540d6bb7838bf63a0f02f573fc48ca4d8de34bb0cbfd514f09", # fp8_e4m3fn
-1121 "8490f7a22615c20651a63dbe7b4241929826a4de20292dc8e63bfc3c61e3654f", # qfp8_e4m34n
-1122 "d8720addef2596fef86b1b22e4b62875c9118779ba8723759a75dfcbc649ffd5", # mystic mlx
-1123 "7d0eac95abe8daae454bcd3d166b8bfc6a35fe68278f97479d62dbb6850f38c0", # mlx flex2
-1124 "ceabd6f71c7112cfaa4dfca8711dda97b79fb9b25983f1c95532de226045f1f8", # mlx jaguar q8
-1125 "49e139f50824fef40908ef4307c851e7adaa8b91bed44054c4829600dbedfdda", # mlx shuttle 3 q4
-1126 "211ade1d474f5dc83190aec8be5c4baf52643777790d64de0cbd84f63613e5e9", # mlx flex1 q8
-1127 "7894547154ba3fd6e364e66e2951ee82b4c3fc1ae0f95df6a4f9d1c5a4e98f17", # DeepFloyd/t5-v1_1-xxl sft
-1128 "eb529f693f4b17773a24e787fcba29486d5e1700dadcc20bb91e4c8b00212d08", # pixart a
-1129 "d80116f6fc39801e4eef425a584e7a7a41cbe5119797bef2dad67299909fe2ae", # Q6K
-1130 "31ebe18e901bfb6e5709a20ec1c95fce29bce2b9545073231e0f909a53239f5c", # Q3 KS
-1131 "6be2b0b7e2de7cf2919340c88cb802a103a997ce46c53131cec91958c1db1af4", # Q4 KM
-1132 "b51cbb10b1a7aac6dd1c3b62f0ed908bfd06e0b42d2f3577d43e061361f51dae", # q5 k m gguf
-1133 "9ec60f6028534b7fe5af439fcb535d75a68592a9ca3fcdeb175ef89e3ee99825", # q8 0
-1134 "8f5ab879234384235d56732f0cda07bf8801f30a49645248c5bfdeeb1665f64b", # q3 kl
-1135 "86427a1f4dba48940e45bf78d6db5bf0d48fce8b4656f5aba27955f06af9628e", # q5ks
-1136 "88b696cfae098f03bb078cc5944ef03aec1e91ec020a6b016b723a0f0532558c", # q4ks
-1137 "1dc600961d3c5ed081f6700485cdc7ed9cfb4631f2dc385b7ac6bd3c80846d0d", # f16 gguf
-1138 "f28631189911f8d7931e8fe642a4cb2a3c51f50da7cabbfa06b89bafc19c00d0", # q3km
-1139 "de9dfdd19d7ba6859993cadec5100665dc7a4fb71e1c6c8970959cbdaf4366e3", # f32gguf
-1140 "7a68b2c8c080696a10109612a649bc69330991ecfea65930ccfdfbdb011f2686", # allegro
-1141 "2c0c539ab8e8fba3877cc94bc483e427f74c525f817a809b028ebc8d96d75a94", # hyd 1.1
-1142 ],
-1143 "layer_b3": [
-1144 "ca94e03b7b1fdcb0d6ff5205eac56f145d2dff8a9c489faf80935bfec8387f18", # bf16
-1145 "c0e2b054bedd782909191b05748a88c28d1538fa91789fec63f036ba01dcc001", # fp16 sd35
-1146 "672de9b79d14001de7d1109ffc52e4d0cccc3bfee6f45648fa347703b58e2b99", # fp16 sd35 diffusers
-1147 "abdb187a996c51cb0469630c124b14eeb0bb8f5f635aca6c71dea264f8bd61ae", # shuttle 3 aesthetic diffusers
-1148 "8926f862b7763fd9688af317eba7809aa71a478484be0c738c269de368ace4a7", # diffusers
-1149 "e616b754cf55e55b3f9f17ab7e1fff95f0607c81782822fc1223ae22fb1e9f36", # fp8 e4m3fn
-1150 "b79e5f1878a62cd726bb4f9fc1415cacb071d278440e9026290c7b36cb41e1d4", # fp8 e4m3fn sd35
-1151 "77619d5278d9f547ddac17d4d99df56cb6a3a9e660ae31b2f896a4297907e62e", # mlx t5 jaguar
-1152 "c87c9d3cc7becc46ee34821299cf8551a6df5541582a45469a031bccdc4bd340", # mlx shuttle t5 q8
-1153 "7e6c32c01c89fc5d1610c410135aa9708e77a7444510e5e479fa677ff2b53643", # mlx jaguar q8
-1154 "a49c2bc301733967ddff113790e301773dc5dd71368b657af4141458de593ced", # mlx flex2 preview
-1155 "c2ea94030ea362e03d73d448fa5353ace0a449dc38c51a4a49fb148444ebb8ef", # mlx shuttle3 diff q4
-1156 "4a90463350f08ef41479da1d561ab41b8f8b792f1603a092226a838156aebfb0", # mlx flex1 alpha q8
-1157 "f86cd0324eebbffb81b15ad47dc8b63fedfa51dc222e44e1a958a7becce2bcb0", # df safetensors
-1158 "48c54c61c5f14e42761c6177539b2da3a22222516dab053952ca8d8e92f93d65", # pixart a
-1159 "311332d9738773669128814d944b1e860a8e3176b37abf43370bc06b43b454d0", # flux
-1160 "3f4e51dec6d542759cdea49b3bec14c090a4908f953fa3e182e2ea43b5b05402", # q5 k m gguf
-1161 "beb25461e168359108add77263ea5cc121b7584cc4aa304ffc4e134783bb1d88", # ggufs
-1162 "43313f90a359c8c1c787a7a833b1ab9f7a38204ba36d0ba587c658d0d9bf0852",
-1163 "fa9e97cdad26f55fedab83a3f114e0338c9cca3ea2bf8f1b168a6dfc5919bf8e",
-1164 "93108d67f8829a7e1e8f3773e9ce53c67f365889c2acfd69816ac80fd43f8e08",
-1165 "fc65a6cc55e89394d7bc0fa4ee952d63ce3bdc143b84b5aa4bb3edf7722a6b83",
-1166 "8163bc781a7e013dfeb806bbb828a36913cf119363ea5fcd9071d87a0c227cda",
-1167 "ad2ba63e1134bad1b15ee339313bc130708b2995e8b4b76fb44d727f28c26ad9",
-1168 "4a844772638ffed2f61d45eaac984094b92540fa1391a4098608fc73a6cd4fd8",
-1169 "76c31e1fd35da7de7cee97c1e7c5ccde640e6fac3e17a62e115ecf484c7196c3",
-1170 "a4d672e22b5bdd8f8b0885cec4a173d0466bb1dcbfbf8400cedcc41c2494f16c", # ggufs
-1171 "d1860c3f01dc9f260d98b50d3d2bbc8dc2d3eefaa93778a8de9d7adfb897fc6e", # allegro
-1172 "b8719092fc58487406211f52dc55bf40b573ccfd29933a989c33a36b694f6f0a", # cogvideox
-1173 "795e272409bc4fa55f402485acf86b607256f91aa965295c5bb771c61f8e9e74", # hyd 1.1
-1174 ],
-1175 "layer_256": [
-1176 "bb20f7805209379aea4d6548f17e551cf27d0f8426ca169e4df8234f718ed5ef",
-1177 "431580c2d86f9a9ed3500f776a4c997223e5644aed211f965354869ccfa4d76e",
-1178 "2ccd548c4ffe34168c60779ebd497b9b410981a2fda813c8723a24a805c94ea0",
-1179 "a608fc4e1cc9762e46187a1ce66e98e8ba4bc3a604cbfd96174bd876baea0fa1",
-1180 "dc9e74cdf535e0b7a17e1335d0d8b38a00f94facf0cb01363baee09945a25278",
-1181 "f07409710a69b2247aa4723a9b40d2225d5e5bfba7b60c51f0ea901fc2ef5ad9",
-1182 "ed28f8b6cc472f352fc840b5a9f841ff17d76ae6918f0676464dca20529aa92b",
-1183 "97c1a08f87c59b4c55ad4672841977cfce43ca7730bcd11d8c178a9330de1855",
-1184 "968972839b859a9c4457f190fad2e17e8585ce27d9ef318df4f5b4e902143944",
-1185 "4dbdeadc957c898c327197a3d8770188535672e9208beb29bbf48dfdf51c8955",
-1186 "669172c2b5e8b97774d9dd0227ede40c4d25cae3adae97d9f281d03531e7e137",
-1187 "39fff130b9ee240102c28a78ee1c4a643e9f800b734ff133f3ab2ad1357bd2f6",
-1188 "6e047ed8cb7007034ff15840dd53c92096f0e7ed5befa07808de8afa35d35874", # safetensors
-1189 "adbd0baa059074501b7686db2b0c01715f3a317275c2657c5dfbfd6ee92389b7",
-1190 "eb63790fb32b5660de34fa42c2e608df58f7aa3680b4984f0ee9008fe613729c",
-1191 "f125c20a33b0ff2dbd4e8ad9acebc34383cb2ef98668169ef79a8c06655ced35",
-1192 "e64e0ac83a785ef584a0e86b347fae8f9e2bd84324a49396ca8a9fe7532a947b", # GGUF
-1193 "70001b3ac1b66522142bb86e4c3e87e20c2bbd07276c763878e0838ef6184aad",
-1194 "f46fd1e2b5fef3b9f7ae80d183cc77f7be181117a72a0bb933bdef0bc6cd679e",
-1195 "83676d73726d101325a47c7f8a60cedf10bab99ea79a6bedad7761220cb4a625",
-1196 "a621a907586e5e270e7c7873b167364d8a935ff347d8240fa9bab319678da690",
-1197 "f0af1a089f40d8611db5c59469314f1547e2df23c6eff24860359b37ea9bd966",
-1198 "72478320b8dbfd9aeaea010dcf0896e3116fa5ab940f3b472882d9f9d2d7333f",
-1199 "9c1a88e36334a48d8482fec54b14ea1d5fd31f0dbb65d13cc616e63dc7c42be5",
-1200 "d0689f727e8ac4fef3ec4b1f29e8a3bd12e1116559eeefb2a1a457cd4e676d1e",
-1201 "fea158a4afcfaa6e95e04799bae0287de0c4fcb188f3b41768a46ce48c71c9df",
-1202 "2e5bc4e73312b5aec4c1a55631cb4ed69cf34ccaa6d1f28f7045f137a579b439", # cogvideox
-1203 "015fdecbc3b5369dbcb2302e4b79985437ac4496d1b9ad63316423a222fb0803", # hyd 1.1
-1204 ],
-1205 },
-1206 ),
-1207 (
-1208 "google/umt5-small",
-1209 "UMT5Model",
-1210 {
-1211 "identifiers": ["encoder.block.1.layer.0.SelfAttention.relative_attention_bias.weight"],
-1212 "file_256": [
-1213 "a8e861969c7433e707cc5a74065d795d36cca07ec96eb6763eb4083df7248f58", # wan t2i diffusers
-1214 "decf9b70814ed5e9965bfca9fbd0483462e2bf743790663025b7742f8c014c72", # fp16
-1215 "0a07449cf1141c0ec86e653c00465f6f0d79c6e58a2c60c8bcf4203d0e4ec4f6", # auraflow
-1216 "c0ef3a140898e228a3520c9adec60743d2e8e5b3d229651bb37f1a3921919f99", # wan
-1217 "7b8850f1961e1cf8a77cca4c964a358d303f490833c6c087d0cff4b2f99db2af", # wan i2ixxl sai fp16
-1218 "c3355d30191f1f066b26d93fba017ae9809dce6c627dda5f6a66eaa651204f68", # wan i2i xxl sai fp8_e4m3fn scaled sai
-1219 "fa1d36fd54f171ae60fea915c23bd77986b330bbed9729f0d2f8ecbe9168bc48", # gguf
-1220 "4a3176f32fd70c0a335b4419fcbf8c86cc875e23498c0fc06f5b4aa0930889e0",
-1221 "adbc782b9145a27e15d63dfa25057efca0ac75e2db7d372c901ddaa130ca2def",
-1222 "b7e2ca4c493c9d51fa951005e8ceba2f4b6b6877cfb4c36a8955c6cd68a1dba7",
-1223 "2521d4de0bf9e1cc6549866463ceae85e4ec3239bc6063f7488810be39033bbc",
-1224 "9209b4c77b34ad8cf3f06b04c6eaa27e7beeebb348a31f85e3b38a1d719b09ed",
-1225 "8bc12d80bc0413573fa58a93626117440b4528f640dd9cb310732e05fa9e6c3e",
-1226 "f64f8d6dc4d8a24276df69d0ccea789aae686f7417950a41e6568c30cb478a5c",
-1227 "17cf97a5bbbc60a646d6105b832b6f657ce904a8a1ad970e4b59df0c67584a40",
-1228 "eaea358bb438c5d211721a4feecc162000e3636e9cb96f51e216f1f44ebd12ce",
-1229 ],
-1230 "layer_b3": [
-1231 "cd92b29c9099a640e3f5d4a76e64b3467f87f6c056119e0defdff94d311ad6de", # wan t2i diff
-1232 "1c943dbcb8b328a7c6c852921ddaefbd84c9df8c83bc51fe303c1f06cb734102", # fp16
-1233 "1639a6467af0db1e15828d33b878e568cba1335947eeadd481170bcdc9ba8e33",
-1234 "72a0329740dee29a2c099eec3c320b3945590a74293356014c30249fe69652e5", # wan
-1235 "0374cba03c607ffe8ab8f04994d82f82e80901dc7578f1a9a6cb2637608be5d5",
-1236 "d75a407f873e1cfa1a0a36214b53b14bfebe9253ea263465151c07f0d57f3f29",
-1237 "621153502b985c143d304318c91dc3d10296d24268c81e3538fc336fdc84c915", # gguf
-1238 "43bb052945d38a68bec27c3d26162e88e306e6074d027d3b4b2b8ae2b1851691",
-1239 "98f50ea5d55e61c1478df47e567e48bdd036d240b9129e64d53a826406900adc",
-1240 "9400313b8eae31699473daa5f840d25a4ef660f68de9a7894f1a28f214f23384",
-1241 "9f13826b8e4ddde24d80de6a947a7868e26cea25dda52790ee6ed695ff72b9bb",
-1242 "475773ab108a537ff904b84e7f3a80129ba4983deb7170b6b52c922ece6069ce",
-1243 "5ef27b3c1eddb08cfe41b452cf9529d86dff811645d40c165bae324486d19e96",
-1244 "e170559d8551cfe651344594e54c0a9a90c0068b00f3866f6e9a3737e20925cb",
-1245 "e8dc7442a20bcdc7b6e5dd0265939d88896eab5ddd33ee16f1f09537e65914b8",
-1246 "4d3d5049857d01741780daf01e96617092973305637b435f4895499a26bbaede",
-1247 "7a2adadc2372feda23b2169337276adda6d1fdef82ba69f0d3321c4c6ba8c604",
-1248 "0a7c61a85bb3f51f75924de48ef3f5e87cbf8901f600cbfcae97f5e2919c4148",
-1249 ],
-1250 "layer_256": [
-1251 "467916d35f3053dce1d40d998fcaf6aa03feda75aa578d964dd61461e23641a3", # wan i2i diff
-1252 "58deeef888d4ded4ffababfbf8da27227a4a6ff8adfa42016e12c0180f713816", # fp16
-1253 "178ebd3fa3418d33a2e45a80d8b9d3662ff4a8e75f3de3f0332f82c505d8152a",
-1254 "8700dcb651465fe6c925b7ad6068b58b32951832fff0ed19819510f8d0713ee5", # wan
-1255 "954f2129ba166e746c71433f717b572d8869ec14b32b7f214d1701d3b1120047",
-1256 "32f5fc1daea014b6488b96c2a1330e0aad87e074844fa3e2e3f20b9e58440395",
-1257 "9245abaf6df8a4b5fcc828ecbcd7b21a1b19bf5f3c4388fb5c8eabc140276dce",
-1258 "172d0fbbd379ae014a7008e148813818494e9e645db802fd000d443369df9d17", # gguf
-1259 "2fa68a26b0386aaf9123d2b4067dafc8631ee724602197dd353f3ea5a61dac8a",
-1260 "16f0054014e6d07b86b0526d5bcfed7d2aa3aebe3e44e6758933d90cbd3da46e",
-1261 "fd62047f5d27ff43210c117dc0f253c101e694a5331d6b684688606c92c65ccf",
-1262 "ddc4f38db9f132fb1b736c1d693b5c039a2d6fe83bdf4f1c1e7a2745b5d79124",
-1263 "9e9ab11b3ea059b84ae2bcc5be76ab3f730a486d92a16f1fd2a959bdc2ede08f",
-1264 "bfb178b1ce27f00e122d2328c662fdef6cc239c07efc749aa61ae2d395441b02",
-1265 "50addf6a911b90194a75b0212429d1af55eb2f9d24715479b9ccc4a40adc299b",
-1266 "2e46e9f1b714d72160d3b3b775a845b3049a01396fab935f1278d9e8de2ef0c6",
-1267 "db8d2b49d9042e39d6531b33ec3bebb9cdf42b9e6ad56163f08da2a7da2a53cd",
-1268 "2d81d19ad5440422b85e0b17c71914269f6c25c9b1fa321c0dd6119ddb41d62d",
-1269 ],
-1270 },
-1271 ),
-1272 (
-1273 "google/gemma2-9b",
-1274 "Gemma2Model",
-1275 {
-1276 "file_256": [
-1277 "e909230aabafad02d097c7dc02f2ae062b4e6b0593477c1f07679d277e09ce71", # sana bf16
-1278 "d61628bc793240439e608c5ae744f55ec8770f684abb63602648a24cb6da60bc", # lumina 2
-1279 ],
-1280 "layer_b3": [
-1281 "55a3c812ac0832d154867f5927365bcc776926e48e65f7f35a81fc11f4bb81da",
-1282 "543572889beb25cad83a43ce70cdd255d2c82951d6595e8c97ff62fd05871c99",
-1283 ],
-1284 "layer_256": [
-1285 "a0d820c39578cf888f398579d9a00d69b31c81e049795ba70008dad8fe5b3a33",
-1286 "abc83b04a04467579ea1952a7efbdd252b8641ac0e2a6a9be2a5a73e371111d6",
-1287 ],
-1288 },
-1289 ),
-1290 (
-1291 "google/gemma-7b",
-1292 "GemmaModel",
-1293 {
-1294 "file_256": ["01676b4c6e765f737a5e9854a315de3887e939c370cae116d505777729099a68"], # lumina next sft d
-1295 "layer_b3": [
-1296 "438d82c867240f194a4e15798eef2886a911c8f57fa2d9f4ffad1d56e7bd1ccf",
-1297 "1de38e09f5f2c5345de48b8cd4dddcfff3e341cc0059752446e186b3863f0981",
-1298 ],
-1299 "layer_256": [
-1300 "e4835a72d582b4ae066d6ff0519f2ee9f8b21fb02e8c28d8eaa317f8d1e9ea75",
-1301 "1657c7180b48672004f4463308dfdd56d92eedeb23d1408ea766985ca208e5aa",
-1302 ],
-1303 },
-1304 ),
-1305 (
-1306 "google/mt5-small",
-1307 "MT5Model",
-1308 {
-1309 "identifiers": [[250112, 2048], "text_encoders.mt5xl.transformer.shared.weight"],
-1310 "file_256": [
-1311 "0524484ec81425ba9deef6fac1393a78ba9b1c9bfed704a4be5f9c7255975cc1", # fp16
-1312 "32f70f1d187e131a5fc3e4f0edc97ce89360d8e2f1d90177a443a05296097acc", # fp16 enc
-1313 ],
-1314 "layer_b3": [
-1315 "a1d616c37711ec7b9073d04734af2f5fd02f9035a322eb46efeace922e104c51",
-1316 # "bc71d4259f4feaa0fb27c1f288765004840f39247cddc98b3ac37329ff1354d0", # fp16 enc
-1317 ],
-1318 "layer_256": [
-1319 "bd337daf0c1aa36896013109b406a0580aa3bb8ab9291d89df3015d737358e95",
-1320 "2e40c48c96fc7df636aad96d3e78ed0ba9f68c3059e21b7fcf917f284c569a61", # fp16 enc
-1321 ],
-1322 },
-1323 ),
-1324 (
-1325 "Qwen/Qwen3-15B-A2B",
-1326 "Qwen3MoeModel",
-1327 {
-1328 "file_256": [
-1329 "c56947057481fb5e7cdf766e442da81717b34addc88bbe8f3728fd25bd03cbae", # qwen3 coder 53 a35
-1330 ],
-1331 "layer_b3": [
-1332 "d2d1e0875202f5c9c84c781a2105620250733bd01832f67b2c17bc981d1eb508" # qwen3 coder 53 a35
-1333 ],
-1334 "layer_256": [
-1335 "408c01da57c4968b7b0e36d98a74e321153e7aeb058fea63ffd140e323526476", # qwen3 coder 53 a35
-1336 ],
-1337 },
-1338 ),
-1339 (
-1340 "Qwen/Qwen2-VL-7B-Instruct",
-1341 "Qwen2VLTextModel",
-1342 {
-1343 "file_256": [
-1344 "1f48ac458d6fbd0aec53a116065a7ee3f1d34bddde544e25c16a05c9d5392b78", # orsta 32
-1345 "0e85c7111ce849293e97aa09ce1172352ecece023a3ecea7ac8311e326b47f3a", # orsta 7
-1346 "d725335e4ea2399be706469e4b8807716a8fa64bd03468252e9f7acf2415fee4", # qwen img
-1347 "e10bd9583a77250376d9134cd6b46799029dfa3b4d7989c1050b3ec149cc7cf5", # qwen flux
-1348 ],
-1349 "layer_b3": [
-1350 "e4f681bde70a753f30f83495a2aa340d251bf3d818eb5a1cbe58f85fd6ea0d40", # orsta 32
-1351 "47b062ce8ddb14845fb1a71d2fd88fd52a82e26561ba3eb05be057915a867775", # orsta 7
-1352 "b6386f70b528ffa9e09fdd8db8a7b91a7c462ed97b06963576c6139e25fdcf31", # qwen img
-1353 "4cd449df9f9004a7e53005583a7e4cfa6de42912f03647d2ea799d489e9c1406", # qwen flux
-1354 ],
-1355 "layer_256": [
-1356 "ed36a4a11c4ebebb10d1e010cb93e2e43fcaf975cd42bb6c9958537593d0d44d", # orsta 32
-1357 "f7f6f64e7b6d7826400a2fc0eef942a47c47bd5914e051ad0c8cd9ff5ff7982b", # orsta 7
-1358 "f341ed0f792cf0570ceb21d3b64ed14bf9875e9fcb90116851364eeed683a6ca", # qwen img
-1359 "ba031d0da78afe24ae63558ad29b8028244a7bd4750a5615dab9079fe32a5fd7", # qwen flux
-1360 ],
-1361 },
-1362 ),
-1363 (
-1364 "openai/gpt-oss-120b",
-1365 "GptOssModel",
-1366 {
-1367 "file_256": [
-1368 "68a8dc1f8e2e5996cb702f14332a25ddf3463daeab2df68e21ca09ef181203c3", # original model
-1369 "a881aa5f561b26a22b14a8262aa61849ace349ffd73d74769e030ac90a1fcf8a", # diffusers
-1370 ],
-1371 "layer_b3": [
-1372 "b52807536902cabbf84f99e4fa2f8713fb4ef77e739f06367ee0d486e3222faa", # gguf
-1373 "43c618018db1fd6e915dead610652da261d9058b73bc5355c85c6ac69af4d913", # "original model"
-1374 "ab27ce7391b7fbd6ce3c319faa119afdac68f746af6a0ce2c3400a132f36f6ac", # diffusers
-1375 ],
-1376 "layer_256": [
-1377 "de5dcad822be5ed6196f0f3f6965739993118d14db97b33a94a269f4f1b7a363", # "original model"
-1378 "575f1977ed42d95a050e13dadaafc05a6d94c8aadca8364dca8a62aa4f2b146c", # diffusers
-1379 ],
-1380 },
-1381 ),
-1382 (
-1383 "microsoft/Phi-4-multimodal-instruct",
-1384 "Phi4MultimodalModel",
-1385 {
-1386 "file_256": [
-1387 "bc703090b63eda16f639fa4de7ac54635c23105ab1da2f6ec4d3403151d38ee6", # mini
-1388 ],
-1389 "layer_b3": [
-1390 "cf4add4ada6082f448788eaf2937f645b5212db88e06ee81475b8be0e99063dc", # mini
-1391 ],
-1392 "layer_256": [
-1393 "7ff992b780b2f8993dd6bb9612207943638b2a42badc976ce80893bc205e801b", # mini
-1394 ],
-1395 },
-1396 ),
-1397 (
-1398 "laion/clap-htsat-fused",
-1399 "ClapModel",
-1400 {
-1401 "file_256": [
-1402 "c92b5a2bee69ff5dd05820d9e0a5cddbc9c9b9dd19a6cb3214f0cf4f29a4d1b0", # audio ldm
-1403 "ae69f555e7f1a2333b8e684c9fa8233f44a47bbadf76d484f941b74f74d2753d", # music ldm
-1404 ],
-1405 "layer_b3": [
-1406 "a4d26450ac399d51b9abbe37859615bb02a5cbf63521da4c7cdc549d04a2872c",
-1407 "ddf310d8eb2d4e3f61e605978675a9d3a748cad9406b9aee8335eae013e77573", # music ldm
-1408 ],
-1409 "layer_256": [
-1410 "843ba86000971d6067bfc4f3ed6dd01bd6f6726188aaa15d86b05554f4fe8481",
-1411 "27529e30442d030a28badf9d62710f4b74e38e9c4424ed169c7e0ac072f5a771", # musicldm
-1412 ],
-1413 },
-1414 ),
-1415 (
-1416 "google-bert/bert-base-uncased",
-1417 "BertModel",
-1418 {
-1419 "file_256": [
-1420 "c6c6348af2cb4d5852fe51102ce39605903dbe7925c005cf8995506cc21ea914", # hunyuandit
-1421 ],
-1422 "layer_b3": [
-1423 "30d7d2cc3ec9e4ba45844e005d0bbcb5887b6a0976042f73da916237dc5c4c12",
-1424 ],
-1425 "layer_256": [
-1426 "94fd2508680ff684eff57e4a5a8ca46bf338fc356a9cf6fe8db2b84543dd7971",
-1427 ],
-1428 },
-1429 ),
-1430 (
-1431 "llava-hf/llava-9b",
-1432 "LlavaModel",
-1433 {
-1434 "file_256": [
-1435 "f5ad57d3eda300a3195bc9c0bb36ab76ebe88831f128e9851e63440aff4a6741", # hunyuanvideo
-1436 ],
-1437 "layer_b3": [
-1438 "d7d6ccb9dbba90b64e4cd259b6309e56708b3f4fbd6e9f85e9f0410e549133ef",
-1439 ],
-1440 "layer_256": [
-1441 "9969c41152aba689413b7f63888ecdc0c0badad2c2960e689ebc4c0e4a696c73",
-1442 ],
-1443 },
-1444 ),
-1445 ]
-1446
-1447 additional_tags = [tag_pipe(*entry) for entry in diffusers_addons]
-1448 additional_tags.extend([tag_base_model(*entry) for entry in transformers_addons])
-1449
-1450 assimilate(
-1451 mir_db, # format
-1452 additional_tags,
-1453 )
-1454
-1455
-1456def add_mir_diffusion(mir_db: MIRDatabase):
-1457 """Create MIR entries missing from the database"""
-1458
-1459 repo = "microsoft/speecht5_hifigan"
-1460 series, comp = make_mir_tag(repo)
-1461 mir_db.add(
-1462 mir_entry(
-1463 domain="info",
-1464 arch="gan",
-1465 series=series,
-1466 comp=comp,
-1467 file_256=[
-1468 "d9dc6513c30a5b86c2497712690c04fe74b4aa79fdab6d490b34fcb4e24c590c",
-1469 ],
-1470 layer_b3=[
-1471 "85b5acdf29ad04c63f885383340d8e3445ae0055521f82cabb82bd09cfb9a956",
-1472 ],
-1473 layer_256=[
-1474 "bd52b538e7ac05711be9321cfb7619d4056996ce32923c9c91ee02cf69154770",
-1475 ],
-1476 )
-1477 )
-1478 series, comp = make_mir_tag("lodestones/Chroma")
-1479 repo = "lodestones/Chroma1-HD"
-1480 mir_db.add(
-1481 mir_entry(
-1482 domain="info",
-1483 arch="dit",
-1484 series=series,
-1485 comp=make_mir_tag(repo)[0],
-1486 repo=repo,
-1487 pkg={
-1488 "0": {
-1489 # "diffusers": "ChromaPipeline",
-1490 "generation": {
-1491 "num_inference_steps": 40,
-1492 # "guidance_scale": 3.0,
-1493 # "num_images_per_prompt": 1,
-1494 },
-1495 }
-1496 },
-1497 file_256=[
-1498 "d845553f11e6afe8139c41ca73678f9f03eab2e68d2e1c6f03ae19509a4d546", # sai
-1499 "1b2993a44e63b2250496f69edce643bac2fb79833cf92ba8dd95cbd764d970c7", # annealed sai
-1500 "2dd46f08516246df1f582047cc09268ce4f747357baff05b13148e71519029fc", # diffusers
-1501 ],
-1502 # layer_b3=[
-1503 # "8da38c3719e77a38a20356c9f92f5ca0101c17406d7a9817323cf67b74088520", # diffusers
-1504 # ],
-1505 # layer_256=[
-1506 # "267798815e0855c2253061c6a6ab70edf9590e8ea1ba9b4621eeb0f6615ee37b",
-1507 # ],
-1508 )
-1509 )
-1510 repo = "lodestones/Chroma1-Flash"
-1511 mir_db.add(
-1512 mir_entry(
-1513 domain="info",
-1514 arch="dit",
-1515 series=series,
-1516 comp=make_mir_tag(repo)[0],
-1517 repo=repo,
-1518 pkg={
-1519 "0": {
-1520 "diffusers": "ChromaPipeline",
-1521 "generation": {
-1522 "num_inference_steps": 8,
-1523 "guidance_scale": 1.0,
-1524 "num_images_per_prompt": 1,
-1525 },
-1526 },
-1527 },
-1528 file_256=[
-1529 "2c0c7d908d04418a48b453c293237a9826d54472cf0ba76e28697d1309d1021b", # sai
-1530 "c88f6794753ba23e8f6bf8c84cf220daa35a6aa16d54ea0c3e0136f52e5da7e1", # sai delta
-1531 "c759d67ca3ef50a9a1c242e3291c57f406646f226a95f43f66577996494986db", # diffusers
-1532 ],
-1533 # layer_b3= [""],
-1534 # "layer_256"= [""],
-1535 )
-1536 )
-1537 mir_db.add(
-1538 mir_entry(
-1539 domain="info",
-1540 arch="unet",
-1541 series=sdxl_series,
-1542 comp="pony-diffusion",
-1543 file_256=["67ab2fd8ec439a89b3fedb15cc65f54336af163c7eb5e4f2acc98f090a29b0b3"],
-1544 layer_b3=["bf4c2154daa4ece7292277b210d081f98759e9ed4d5c889564632e3ccc4a1071"],
-1545 layer_256=["465425d4420dcf5aa4b4d5b456db11a1fcc7c8f61b2e4a87e2470297c98bb96e"],
-1546 )
-1547 )
-1548 mir_db.add(
-1549 mir_entry(
-1550 domain="info",
-1551 arch="unet",
-1552 series=sdxl_series,
-1553 comp="pony-diffusion-turbo",
-1554 file_256=[
-1555 "7555ac941f3a767833830ba5cc9a4508a9777cbf97b487b6baf0400ab7000587", # turbomerge
-1556 "9322f9d91b28abf09e4137bc02ec806af23510221a164e71b81778e61cc3b4b2", # turbosimple
-1557 ],
-1558 layer_b3=[
-1559 "1e8f23fcd4be0f00eb52368b91c709fffa8a3b8e21772b92b2e0671eed9117d0",
-1560 "5c8b3f34f9d0a58135cf72fbfe9b5d75b5545a10e3d726478543fa7cc510a8bc",
-1561 ],
-1562 layer_256=[
-1563 "7edf51ef09b39c46937a4e4141707c040cd12af0d95299a4d3cd2b7d3fabe035",
-1564 "74e4dbc89d57d61ff7e8af8b0fddcf7466ba233d53ca4ffb7777138991bc3d52",
-1565 ],
-1566 )
-1567 )
-1568 repo = "cagliostrolab/animagine-xl-4.0"
-1569 mir_db.add(
-1570 mir_entry(
-1571 domain="info",
-1572 arch="unet",
-1573 series=sdxl_series,
-1574 comp=make_mir_tag(repo)[0],
-1575 repo=repo,
-1576 file_256=[
-1577 "8ece83aa1bed1fb39a2b81f1660f0ce6889218e493c1f2ed55e9f15f59a7e03f", # v4
-1578 "6327eca98bfb6538dd7a4edce22484a1bbc57a8cff6b11d075d40da1afb847ac", # v4 opt
-1579 "1449e5b0b9de87b0f414c5f29cb11ce3b3dc61fa2b320e784c9441720bf7b766", # v3
-1580 "e3c47aedb06418c6c331443cd89f2b3b3b34b7ed2102a3d4c4408a8d35aad6b0", # v3.1
-1581 ],
-1582 layer_b3=[
-1583 "268ffbb120670b9c4b25158bd474c787740884b7738b48203aa03c4c3f00028f",
-1584 "18fda1a55cad137d62c81d4328f5ece85d88b126261e06b9e14ab68055d5d484",
-1585 "bae9bc8a5c43145bcf92ee3391618d9eaddd689f626991bae202de9cf5f1e70e",
-1586 "d6bc5ccafa2b97c867b13a1e7a8c2c7ad9c4877055a66c71bb773557bc306447",
-1587 ],
-1588 layer_256=[
-1589 "c21d1c38813e078817122e12866ab39f5aa7f56945dd4a8beee3cae1e0f139e7",
-1590 "b916c162c981155aaf74e93d5314038af6767bb5a129c51ee05a1fb6a206c6ac",
-1591 "ecc6bfc73824a2d7c3b0ca184854a235859f329c83768f017b07a19a535d17b4",
-1592 "97f6ca05de7fbdae7aacb2427a552f924492176c474a23dd252c192e1c0e9d65",
-1593 ],
-1594 )
-1595 )
-1596 repo = "OnomaAIResearch/Illustrious-XL-v2.0"
-1597 mir_db.add(
-1598 mir_entry(
-1599 domain="info",
-1600 arch="unet",
-1601 series=sdxl_series,
-1602 comp=make_mir_tag(repo)[0],
-1603 repo=repo,
-1604 file_256=[
-1605 "c2a1a3eaa13d4c107dc7e00c3fe830cab427aa026362740ea094745b3422a331", # v2
-1606 "536863e9f0c13b0ce834e2f8a19ada425ee4f722c0ad3d0051ec7e6adaa8156c", # 1.1
-1607 "3e15ba00387db678ab4a099f75771c4f5ac67fda9e7100a01d263eaf30145aa9", # 0.1
-1608 "e3d12d0f76d61aa31d2668a2217e5b642592193f2946842c44d7056ea5469cce", # 0.1 guided
-1609 "735cf3fefcbdc4f7817f53247e38b836ffd27c7641af6d8daa21d245242cb4bd", # 1.0
-1610 ],
-1611 layer_b3=[
-1612 "93b061baf21d743d592327a61f027d099d8e18da9808a76c7704ad123eba4a29",
-1613 "dc05fed2acbc73cef4c377cfa2a681c5cf6d065b88d8bf70d371bbcce6a223a8",
-1614 "8eb1c30327e5b71b35b9a4513dc5f2cac9f244667393c0eedb10a26aa9991cd8",
-1615 "3dafbe31f6ebaffa3d054e1b37049e1147faa2474ceb6dab7bc3c4cded0c845e",
-1616 "892533778ee14454938f7b50830093f58e12f1e14560a148f71927e4ccff5f5c",
-1617 ],
-1618 layer_256=[
-1619 "397791b3d77affb7bd35c5ded7377493c6bf456920a41388ba95bd0157109803",
-1620 "b23c02b8519c6777a1f271662f4251a59468c4b3e11184a2d722fa8929b4ea48",
-1621 "a373981494f5508c124a1960bdd096bbc96935fbb54b1218f563206d3892c176",
-1622 "b709df257c40d9d981f686f2880bbe64f43b78805b7213768d659a142a593efd",
-1623 "f1e6b4cab0fce608dca6fa851384e8728202449f16270fbd1f0c4c5ec4946c10",
-1624 ],
-1625 )
-1626 )
-1627 repo = "playgroundai/playground-v2.5-1024px-aesthetic"
-1628 mir_db.add(
-1629 mir_entry(
-1630 domain="info",
-1631 arch="unet",
-1632 series=sdxl_series,
-1633 comp=make_mir_tag(repo)[0],
-1634 repo=repo,
-1635 file_256=[
-1636 "11b6d7bce65674659cc6b7ea960658436edfd80e566cb240ebd4bfbc3e2076c8", # 2.5 diffusers
-1637 "bcaa7dd6780974f000b17b5a6c63e6f867a75c51ffa85c67d6b196882c69b992", # 2.5 aes sai fp16
-1638 "956dca99114aaa5c3eb526381309d37ee96737e78ed64c8ae613409f47c3f65a", # 2.5 aes sai
-1639 "933778ce76c1fc0ca918b37e1488411b8a99bbd3279c12f527a3ac995a340864", # 2.5 fp16 diffusers
-1640 "5c7d38880d0940e6795158b7608ccef89217272b1f2a9331c5b0a2adffcd82c4", # v2 sai
-1641 "0411e988479884b1a3ecd184123efe38d051d8d0ef24270585a7d1d57499464a", # v2 sai fp16
-1642 ],
-1643 layer_b3=[
-1644 "d55b22740da2d5b98020ad2390cdc0a7ee08cf9e0d98c11957f16cc20c49815b", # 2.5 diffusers
-1645 "7e9be9bd9a3aed1ad7207e2f77c98c24c3a75f6adcc9b53514033c6c3365d289", # 2.5 aes sai fp16
-1646 "5c6dfcc8d01dfb64723f8f5785caa080e2987859c0a050470bfdbe5312be9efc", # 2.5 aes sai
-1647 "703f775c6e48ed5b0eba6e847414f047bcd4adc677dbc1bf221b3ef05b2ac471", # 2.5 diffusers fp16
-1648 "72d4ebe4af61f8a7add8fe36b8acd16602894279fb5a744ad50b5b5bac7067b8", # v2 sai
-1649 "acb757b851db12cdf9d4365a45ee0d6e64afa77ac95583bb82711baf7c4125fd", # v2 sai fp16
-1650 ],
-1651 layer_256=[
-1652 "adb7be228d4ee6e583c3e5ae4ddb579fef64c3987617ce4d4aff3eb7f8d6a3f7",
-1653 "d4813e9f984aa76cb4ac9bf0972d55442923292d276e97e95cb2f49a57227843", # 2.5 aes sai fp16
-1654 "fe2e9edf7e3923a80e64c2552139d8bae926cc3b028ca4773573a6ba60e67c20",
-1655 "bc7021473a04a6de3fe0d0fed600875d852ad1ad9d47c445278f66ce9e8ec7a0" # 2.5 fp16 diffusers
-1656 "fc94481f0c52b21c5ac1fdade8d9c5b210f7239253f86ef21e6198fe393ed60e", # v2 sai
-1657 "a6f31493ceeb51c88c5239188b9078dc64ba66d3fc5958ad48c119115b06120c", # v2 sai fp16
-1658 ],
-1659 pkg={
-1660 0: {
-1661 "diffusers": "DiffusionPipeline",
-1662 "precision": "ops.precision.float.F16",
-1663 "generation": {"num_inference_steps": 50, "guidance_scale": 3},
-1664 }
-1665 },
-1666 identifiers=[
-1667 "edm_mean",
-1668 [1, 4, 1, 1],
-1669 2516,
-1670 ],
-1671 )
-1672 )
-1673 repo = "segmind/Segmind-Vega"
-1674 mir_db.add(
-1675 mir_entry(
-1676 domain="info",
-1677 arch="unet",
-1678 series=sdxl_series,
-1679 comp=make_mir_tag(repo)[0],
-1680 repo=repo,
-1681 file_256=[
-1682 "94762e983e5942056be73c5c1d4464b8ffa1ada500b4fef1267550e2447953ce", # modelspec sai
-1683 "1ab33e37fbb2566c55cd729e4ab79cc2f99cd9d0a578fabc7a2cf4ee47968be1", # diffusers
-1684 "8cfa375669b1222d6fecf470f41b2abb370c76a90ab9568964c4bb15b34ec8a2", # diffusers fp16
-1685 ],
-1686 layer_b3=[
-1687 "2f353c5e6ed0a2c05af00d014e18e65f69f1ce8c48f8eefbf8ad71b34f940fbf",
-1688 "cc34bd3135d7cafc3cb6e3f6e7cb6896c98277bad52877a952ddbd2ffe222e01",
-1689 "b90efdc848f5386d5250b6fb233ce380cf6cc299f497cfa1d2feaef22f87c9d1",
-1690 ],
-1691 layer_256=[
-1692 "029b89ee311110c8f945dbdfc52c1d5daeb1e78c353c38aa3141ec68ce28e7cc",
-1693 "5cdb948e5f3873300679073391d48fc648171f02093d7737d078557ff75762bb",
-1694 "f73afbe43cc76571cb86ebcfced618668a2fb2252b0bc6ba88d6e942bae75741",
-1695 ],
-1696 )
-1697 )
-1698 repo = "segmind/SSD-1B"
-1699
-1700 mir_db.add(
-1701 mir_entry(
-1702 domain="info",
-1703 arch="unet",
-1704 series=sdxl_series,
-1705 comp=make_mir_tag(repo)[0],
-1706 repo=repo,
-1707 file_256=[
-1708 "7cb406ec0662e91570a79f3c4fb8f0ea5325bffe6af5d9382edae838698f72bd", # modelspec sai
-1709 "1895a00bfc769a00b0c0c43a95e433e79e9db8a85402b45a33e8448785bde94d", # a1111 aio
-1710 "0bf1ce6b065a6b969ab02dc8e8fa21eb20ee189b10935c49ce68c77a7e432c1c",
-1711 "02ed8ebd0ed55aec686fcf20946d7a1659a31f9f8d9c3798cd254ba6b67434ca", # diffusers
-1712 "40d8ea9159f3e875278dacc7879442d58c45850cf13c62f5e26681061c51829a", # diffusers fp16
-1713 ],
-1714 layer_b3=[
-1715 "c074dc38e8ec836816b91cbcc2ca17f80d6106de8d196d416ef9a27c8837ee45", # modelspec sai
-1716 "1d6c0216da57fe98e7ad29e9653566725f5b2a87845fdbdcda257b3be817b5f4", # a1111 aio
-1717 "c074dc38e8ec836816b91cbcc2ca17f80d6106de8d196d416ef9a27c8837ee45",
-1718 "89f86d9c846495870416b4945b6a46a517f28405e5bab666feb4057f012340be",
-1719 "535b47e9b70da6494878ca6d45af3f2e201b7f17748432911c12232e586855e6",
-1720 ],
-1721 layer_256=[
-1722 "52267d5d327a2ba92c7a14261a9d081df621b8366819b1bb3a47d130523a813c",
-1723 "b365a3631c6c74532f3a571c84c68e088be35496d35be1e932031713ddd2a2f4",
-1724 "52267d5d327a2ba92c7a14261a9d081df621b8366819b1bb3a47d130523a813c",
-1725 "89f86d9c846495870416b4945b6a46a517f28405e5bab666feb4057f012340be",
-1726 "535b47e9b70da6494878ca6d45af3f2e201b7f17748432911c12232e586855e6",
-1727 ],
-1728 )
-1729 )
-1730 repo = "shuttleai/shuttle-3.1-aesthetic"
-1731 mir_db.add(
-1732 mir_entry(
-1733 domain="info",
-1734 arch="dit",
-1735 series=schnell_series,
-1736 comp=make_mir_tag(repo)[0],
-1737 repo=repo,
-1738 pkg={
-1739 2: {
-1740 "diffusers": "DiffusionPipeline",
-1741 "generation": {"guidance_scale": 3.5, "num_inference_steps": 4},
-1742 }
-1743 },
-1744 file_256=[
-1745 "176871da1d5d2d511a52ae9b0dd70faa1f5d1b7734b7e33ed6b4bffa52050e0d",
-1746 "4b80d37681eaed07b7f5b3825a392da929d1620933ede7c2749ef3613cc53f42",
-1747 ],
-1748 layer_b3=[
-1749 "ff422d1734abf33366e87bbf44267dc6096c5d499e695287c35558174877412e",
-1750 "5ad8034eac6b82d842311437101c52b5d35826ce34994940d9e667e702a0d45c",
-1751 ],
-1752 layer_256=[
-1753 "e5d95de314cbfc49b79479118a1ac0b90fc95ccd6bb1a5c95803996d6cebf8fe",
-1754 "d299e8ea4a605917ab98a4a7330d4d398b4ae295efbf458eeeceb5ff1bd7959a",
-1755 ],
-1756 )
-1757 )
-1758 repo = "shuttleai/shuttle-3-diffusion"
-1759 mir_db.add(
-1760 mir_entry(
-1761 domain="info",
-1762 arch="dit",
-1763 series=schnell_series,
-1764 comp=make_mir_tag(repo)[0],
-1765 repo=repo,
-1766 pkg={
-1767 2: {
-1768 "diffusers": "DiffusionPipeline",
-1769 "generation": {"guidance_scale": 3.5, "num_inference_steps": 4},
-1770 }
-1771 },
-1772 file_256=[
-1773 "a5b04df4072698395387c21e8da0176d03f6557e0c38ff1dd3bf469ebab9d0fd", # fp8
-1774 "a91b46de2055b3511ee87523b57862648856e8c00100161d5b520543a7302755", # norm
-1775 "23a77c86189d5934da48bf44bb871cf80ba99177ffd3fd5272cdecb208c8b8be", # mlx q8
-1776 "d3782d5a8f6e82c6676e8e26d54020934ada589d2aceb17fc5ca604b1bd55da8", # mlx q4
-1777 ],
-1778 layer_b3=[
-1779 "4dd3174edf6b680ce9daf3de643e33ae2c4f09a4d5968da61ea48885f3a193c0",
-1780 "9fdf191b2c58b2a6e190396e12314530593dca4f2a2bee389ec5175da5e52af8",
-1781 "ad203ad6a00d8b1315337e34069e7c41016ea407469a536de8ad6807042017fd",
-1782 ],
-1783 layer_256=[
-1784 "14d0e1b573023deb5a4feaddf85ebca10ab2abf3452c433e2e3ae93acb216443",
-1785 "7ce8d449b32a9c959431ade729b513ee7a6457f11e1c13e3ef04dd8db3494621",
-1786 "9c3395f67a3d844483b77f0ddd5e2ea64b61732fa9d9da19845bb8ae574c1f8c",
-1787 ],
-1788 )
-1789 )
-1790 repo = "enhanceaiteam/Mystic"
-1791 mir_db.add(
-1792 mir_entry(
-1793 domain="info",
-1794 arch="dit",
-1795 series=dev_series,
-1796 comp=make_mir_tag(repo)[0],
-1797 repo=repo,
-1798 pkg={0: {"generation": {"num_inference_steps": 16, "guidance_scale": 7.5, "width": 768, "height": 1024}}},
-1799 file_256=[
-1800 "179d4000e44295f6dfadc0e4ac210146454724d46371b82657200ff9fb5c68a9", # mlx 0
-1801 "48ca85274e3b67f07f70dd84b67725e62395c2f7b188394342716f783ea4c6ac", # mlx q8
-1802 ],
-1803 layer_b3=[
-1804 "91074aaebe1b5f3b2e7755d3c092af7eb240e92a192360690f1033949d3c8a68", # mlx 0
-1805 ],
-1806 layer_256=[
-1807 "3942e6a52dbb0abaf63b031d9c4eda0df47576b51d4c81361978a3dc27b1309e", # mlx 0
-1808 ],
-1809 )
-1810 )
-1811 repo = "shuttleai/shuttle-jaguar"
-1812 mir_db.add(
-1813 mir_entry(
-1814 domain="info",
-1815 arch="dit",
-1816 series=schnell_series,
-1817 comp=make_mir_tag(repo)[0],
-1818 repo=repo,
-1819 pkg={
-1820 2: {
-1821 "diffusers": "DiffusionPipeline",
-1822 "generation": {"guidance_scale": 3.5, "num_inference_steps": 4},
-1823 }
-1824 },
-1825 file_256=[
-1826 "dcbc4f2470b177eed12c7d7515c0e7342515a849ebd31a50c8d8d43913d7bd32",
-1827 "26a7aa64c0798a3549e1d767932da0a7fb82b49f8edcbdcde804a20d9ed1478f", # mlx q8
-1828 ],
-1829 layer_b3=[
-1830 "9906c29933d0c33a6ee8d9712f33fa8bd4b35b46a1c7b565ae48832b757dd980",
-1831 "89c453c4bf99220405687eed984dace4492bdae1b6fb08f3d9629145b1a11672", # mlx q8
-1832 ],
-1833 sha_256=[
-1834 "4eacf27e5659f5dc42f34c407cbe9e1e202290692df754eb68fe913f59fa2941",
-1835 ],
-1836 )
-1837 )
-1838 repo = "freepik/flux.1-lite-8b"
-1839 mir_db.add(
-1840 mir_entry(
-1841 domain="info",
-1842 arch="dit",
-1843 series=dev_series,
-1844 comp=make_mir_tag(repo)[0],
-1845 repo=repo,
-1846 pkg={0: {"generation": {"num_inference_steps": 28}}},
-1847 file_256=[
-1848 "09e970a7b8d1813ea7cacd48f9a944fd223882b137a8f4f3b61d864cdc20bbec", # mlx q8
-1849 "de90e69945c2f4afcb9b6a057ce48190905c984370fce76b16ba3b97d46e2747", # mlx q4
-1850 ],
-1851 layer_b3=[
-1852 "9276fa4805efeb45c08cca32c5b51d490e57a2ce5c15ef476a8e468a509c5cdf",
-1853 ],
-1854 layer_256=[
-1855 "e1afe2f9b1ca55b3c659293cf3237f6b5571f5c4e826bad025ff0f7b54dc34ee",
-1856 ],
-1857 )
-1858 )
-1859 repo = "freepik/f-lite-7b"
-1860 mir_db.add(
-1861 mir_entry(
-1862 domain="info",
-1863 arch="dit",
-1864 series=dev_series,
-1865 comp=make_mir_tag(repo)[0],
-1866 repo=repo,
-1867 pkg={0: {"f_lite": "FLitePipeline", "generation": {"num_inference_steps": 28}}},
-1868 )
-1869 )
-1870 repo = "freepik/f-lite-texture"
-1871 mir_db.add(
-1872 mir_entry(
-1873 domain="info",
-1874 arch="dit",
-1875 series=dev_series,
-1876 comp=make_mir_tag(repo)[0],
-1877 repo=repo,
-1878 pkg={0: {"f_lite": "FLitePipeline", "generation": {"num_inference_steps": 28}}},
-1879 )
-1880 )
-1881 repo = "freepik/f-lite"
-1882 mir_db.add(
-1883 mir_entry(
-1884 domain="info",
-1885 arch="dit",
-1886 series=dev_series,
-1887 comp=make_mir_tag(repo)[0],
-1888 repo=repo,
-1889 pkg={0: {"f_lite": "FLitePipeline", "generation": {"num_inference_steps": 28}}},
-1890 )
-1891 )
-1892 repo = "TencentARC/flux-mini"
-1893 mir_db.add(
-1894 mir_entry(
-1895 domain="info",
-1896 arch="dit",
-1897 series=dev_series,
-1898 comp=make_mir_tag(repo)[0],
-1899 repo=repo,
-1900 file_256=["4236455adeaeb4ed444d63b253ec99805022d17e962ed7261ada9c72ce11cfee"],
-1901 layer_b3=["c1a6f83585398fe452d20596a79a522e2986f4c2c01a40e7bfd787af113735d3"],
-1902 layer_256=["e4a0d8cf2034da094518ab058da1d4aea14e00d132c6152a266ec196ffef02d0"],
-1903 ),
-1904 )
-1905 repo = "ostris/Flex.2-preview"
-1906 mir_db.add(
-1907 mir_entry(
-1908 domain="info",
-1909 arch="dit",
-1910 series=dev_series,
-1911 comp=make_mir_tag(repo)[0],
-1912 repo=repo,
-1913 file_256=[
-1914 "0407108e446a4f57efffc5e7518bc374876af970d3c6068dc4074de0d221c615", # modelspec sai
-1915 "df168ba94d5f96c478b24604a6beedff6189047152190509c73c162ea0d8ec02", # mlx
-1916 ],
-1917 layer_b3=[
-1918 "7f85cdc186896da6965b57d5edb672f08663075d2b207f0e20e328c4034a8076", # mlx
-1919 ],
-1920 layer_256=[
-1921 "5063de856be5365807d12b47ef6919b4ac611a72651739b2b4050e113bed7a83" # mlx,
-1922 ],
-1923 ),
-1924 )
-1925 repo = "ostris/Flex.1-alpha"
-1926 mir_db.add(
-1927 mir_entry(
-1928 domain="info",
-1929 arch="dit",
-1930 series=dev_series,
-1931 comp=make_mir_tag(repo)[0],
-1932 repo=repo,
-1933 file_256=[
-1934 "5d6dce30a266ccbf530c3a3bf253cd5486720a8fb71cdeed556c28304201dc2f", # modelspec sai
-1935 "7acf8771b80a91eaa21566abe8c7d9d3ba33d8688e6e98446827749aee7ca1ee", # mlx
-1936 ],
-1937 layer_b3=[
-1938 "cb3d3edafd81651eefd62894b3572deb02c5304f4b5d4f7ab8654f1fb922ecd6", # mlx
-1939 ],
-1940 layer_256=[
-1941 "a6b9af6efc25fa77cd24046b81ee66fea09a9987d2a8e56ffca9b7a1c9c9c519" # mlx,
-1942 ],
-1943 ),
-1944 )
-1945 repo = "tensorart/stable-diffusion-3.5-medium-turbo"
-1946 mir_db.add(
-1947 mir_entry(
-1948 domain="info",
-1949 arch="dit",
-1950 series=sd3_series,
-1951 comp=make_mir_tag(repo)[0],
-1952 repo=repo,
-1953 pkg={
-1954 0: {
-1955 "precision": "ops.precision.bfloat.B16",
-1956 "generation": {"num_inference_steps": 8, "guidance_scale": 1.5, "height": 1024, "width": 768},
-1957 }
-1958 },
-1959 file_256=[
-1960 "5b0530e8d71b49fa1358f1208047cd789a40bae5b44406c9524b0f0d88f8b246", # diffusers
-1961 "07119c77c3548a1d9eb30923df4dd55ec74914dc5ec81626804dcbe51ce17a5d", # sai
-1962 "3c379381344d2a2b3ee3d7a1bc97f7d1e58fa95c6b5187fb48b3ce446f99f17b", # q4km gguf
-1963 "6b3806cafdb4303ea2638e9e08eb186067b4a46a95ddf344ccdbe56537afaf6e", # q8km gguf
-1964 ],
-1965 layer_b3=[
-1966 "873821614080a98e1ebfe56673bc96c2ac57379720d4ad2f97e4bca317571d48", # diffusers
-1967 "7284d2027523482af9ef47405667ca891cc518bfb6ebf1f1d4666cb0accc8cd5",
-1968 "d938ee5738c73f701760ed18acad274b074d2796123aee3f2eee1328b6c36ea4",
-1969 "c4c40056c2a77959083b5a69a1a4b205caa463ccabde057352c5c4e38b2c67b6",
-1970 ],
-1971 layer_256=[
-1972 "3c324055a1ec6eb4ee0242e344bb2b6356afcbd2e215fdd9d160cda691a72fae",
-1973 "7284d2027523482af9ef47405667ca891cc518bfb6ebf1f1d4666cb0accc8cd5",
-1974 "d938ee5738c73f701760ed18acad274b074d2796123aee3f2eee1328b6c36ea4",
-1975 "c4c40056c2a77959083b5a69a1a4b205caa463ccabde057352c5c4e38b2c67b6",
-1976 ],
-1977 ),
-1978 )
-1979 repo = "Wan-AI/Wan2.1-FLF2V-14B-720P-Diffusers"
-1980 series, comp = make_mir_tag(repo)
-1981 mir_db.add(
-1982 mir_entry(
-1983 domain="info",
-1984 arch="dit",
-1985 series=series,
-1986 comp=comp,
-1987 repo=repo,
-1988 file_256=[
-1989 "",
-1990 "",
-1991 ],
-1992 layer_b3=[
-1993 "",
-1994 ],
-1995 layer_256=[""],
-1996 ),
-1997 )
-1998 repo = "OnomaAIResearch/Illustrious-Lumina-v0.03"
-1999 mir_db.add(
-2000 mir_entry(
-2001 domain="info",
-2002 arch="dit",
-2003 series=make_mir_tag("Alpha-VLLM/Lumina-Image-2.0")[0],
-2004 comp=make_mir_tag(repo)[0],
-2005 repo=repo,
-2006 file_256=[
-2007 "dc6cffcfb0ccfca6332ddb5d2fe25bcb5f496f44b481627f48c42626156fa6a8", # 2b 22100 ema unified fp32
-2008 "2ac549741fa1c6de2d6cd8be06abcdce52d472eeae2439f948e285258b66a214", # 0.03 ema
-2009 ],
-2010 layer_b3=[
-2011 "a97b4a63e1e7678e8e7154fae55252267bd1f0ba76b03dba622d801644e657ac",
-2012 "aa6c1b2d1971cea3c4ed0963c8d68d4c50db683f8eab9f77f60ea2d04ed6ce5c",
-2013 ],
-2014 layer_256=[
-2015 "39086c199b9ac296dcba53461ba1e113906d91fbc1b12556d92f5cc77ca11f9f",
-2016 "e51ba2ded40f1af5ca6f78c46eed8305fbd87cd6401e9d439837e10d35cc5828",
-2017 ],
-2018 )
-2019 )
-2020 mir_db.add(
-2021 mir_entry(
-2022 domain="ops",
-2023 arch="patch",
-2024 series="hidiffusion",
-2025 comp=sdxl_series,
-2026 pkg={
-2027 0: {
-2028 "hidiffusion": {"apply_hidiffusion": {"timesteps": "StableDiffusionXLTimesteps"}},
-2029 "generation": {"height": 2048, "width": 2048, "eta": 1.0, "guidance_scale": 7.5, "num_inference_steps": 10},
-2030 },
-2031 },
-2032 )
-2033 )
-2034 mir_db.add(
-2035 mir_entry(
-2036 domain="ops",
-2037 arch="scheduler",
-2038 series="align-your-steps",
-2039 comp=sdxl_series,
-2040 pkg={
-2041 0: {
-2042 "diffusers": "schedulers.scheduling_utils.AysSchedules",
-2043 "generation": {"timesteps": "StableDiffusionXLTimesteps", "num_inference_steps": 10},
-2044 }
-2045 },
-2046 )
-2047 )
-2048 # possible mixed-type architecture?
-2049 # fusion / united / universal
-2050
-2051
-2052def add_mir_llm(mir_db: MIRDatabase):
-2053 base_arch, base_series, base_comp = tag_base_model(repo_path="facebook/chameleon-7b", class_name="ChameleonModel")
-2054 repo = "Alpha-VLLM/Lumina-mGPT-7B-1024"
-2055 series, comp = make_mir_tag(repo)
-2056 mir_db.add(
-2057 mir_entry(
-2058 domain="info",
-2059 arch=base_arch,
-2060 series=base_series,
-2061 comp=series,
-2062 repo=repo,
-2063 pkg={
-2064 0: {
-2065 "inference_solver": {"FlexARInferenceSolver": {"precision": "bf16", "target_size": 768}},
-2066 "generation": {"images": [], "qas": [["q1", None]], "max_gen_len": 8192, "temperature": 1.0},
-2067 },
-2068 1: {"inference_solver": "ChameleonXLLMXForConditionalGeneration"},
-2069 },
-2070 identifiers=["model.embed_tokens.weight"],
-2071 file_256=[
-2072 "6b71408a7c574d98f00114ab770ac6addc71471770456e482e7b5ec641c02345",
-2073 "1d5d8d5532bae0f32ba35d10d411e506d61e4378dc9fc338f2b1e6af2aa322ec", # 768
-2074 "a8fe636bbee30fef06dcd8e806ffc65b2aed0ad08a07fdc62f35717d0f851be5", # 512 multi
-2075 "6420fa13483576d46263996627ba7add2237a01f46dedd3b7750112c0cc2d95b", # 512
-2076 ],
-2077 layer_b3=["6cd6b3caaea270feb5aff8e9fec205a27da4f48a1e740e63dc9a08f16e70a656"],
-2078 layer_256=["eaa882db6a69cf8ed0104a15b2cdbbb570a23a06ab8c8f65f4c6c21719c6ba25"],
-2079 ),
-2080 )
-2081 repo = "openai/clip-vit-large-patch14"
-2082 series, comp = make_mir_tag(repo)
-2083 mir_db.add(
-2084 mir_entry(
-2085 domain="info",
-2086 arch="vit",
-2087 series=series,
-2088 comp=comp,
-2089 repo=repo,
-2090 pkg={0: {"transformers": "CLIPTextModel"}},
-2091 identifiers=["text_model.encoder.layers.0.mlp.fc1.weight", "clip-l"],
-2092 file_256=[
-2093 "cb0cba1ead482a850532ebe5ff6b5c8d4456aee32a5228acf0a31e7d9472415e", # long vit best
-2094 "39e79c916feca4ddf546d9fe923e664714b59ea61074f7228037d17c302f3d17", # vit l detail improved hit gmp
-2095 "893d67a23f4693ed42cdab4cbad7fe3e727cf59609c40da28a46b5470f9ed082", # flux/shuttle 3 aes
-2096 "778d02eb9e707c3fbaae0b67b79ea0d1399b52e624fb634f2f19375ae7c047c3", # playground 2.5
-2097 "660c6f5b1abae9dc498ac2d21e1347d2abdb0cf6c0c0c8576cd796491d9a6cdd", # playground 2.5 fp16
-2098 "71e183d11db0c6b6282a4d9e0abb74125edc8692393e89ed8ee5571005f35cb1", # sd3.5 fp16
-2099 "5c3d6454dd2d23414b56aa1b5858a72487a656937847b6fea8d0606d7a42cdbc", # sdxl diffusers
-2100 "87c1c0b0894c9e9e10b962e597e8d64dd3a3a2d372c389922b335a53c250b2ae", # L
-2101 "bd289dd57fee86bc8816b55919a2b03f9c3c75af6025e21777325a6730872325", # jaguar mlx
-2102 "8377b1ca9d88fe06ec483dd7b3cfc62e5e8dbf8ddd252f455e79d659fa0553c5", # ssd-1b
-2103 "5487ea0eee9c9a9bff8abd097908d4deff3ae1fa87b3b67397f8b9538139d447", # ssd-1b fp16
-2104 "92b998a9a64549bfa05c019bde114be6681549a0c79caee903fe30c9444d08b9", # vega
-2105 "1e090d6a828fd92401be5f83e615fd7b4fb1f4a22e9af9040a38f602e839317c", # vega fp16
-2106 "11807cb2522cfe99240e5ee2bbeb1ccb42cecca2215102ee872567c7773b28b9", # flux
-2107 "d008943c017f0092921106440254dbbe00b6a285f7883ec8ba160c3faad88334", # sd1
-2108 "77795e2023adcf39bc29a884661950380bd093cf0750a966d473d1718dc9ef4e", # sd1 fp16
-2109 "b70c11ad5d7e9abf6109348908f599ea382f8019e1f36910bbc8ebecde936633", # hidream i1
-2110 "fc42badf529dd83f2f7c3d20fe6bda1e22036162f37c4c668b9e130884e20561",
-2111 "e27bafa0b3029ad637ef3ace24ce1efe85b8d0dbd22e03a2e70bda6fc88963a1", # onnx
-2112 ],
-2113 layer_b3=[
-2114 "f58a22a381f79985b6d38782f6110a52c2f319b40fdedd3b88b24945dfcbdf64",
-2115 "8faa00b8fd1dbd9286a7237df18caeb8c91af100a6813849b6bae272a01dd7b7",
-2116 "ab5bebc98299c155251a06deccde599ba0128038ee3ce021e8c59a45f58f72c0",
-2117 "c70e9d86a9dcbbbe7c269ef9dfac96ce9c96c46922577338cc1902e5fe936315",
-2118 "f285e9b7b70745df81adc8b558ec74b536b79b6fc02a453ecc61ea9d13f25f1a",
-2119 "7ab17bfa06ab8d65840997ef641f3f593d096860e20141f1eeb0169d131c1c23",
-2120 "2737d3f327e8176dbb549b9c5c4994821430a6c3b07e3bbc925d97511c802636", # jaguar mlx q8
-2121 "58a826a4a5fe555b4df188a1ebc0d8d9c96cedae3a26ce84c247861dbb93388f", # sd1
-2122 "1540fd8844898960e18ce8fd153e5f21a8c446bd8c4d6f536a7cf11418f02bf3", # sd1
-2123 "c4c9caccdbec12b965d93688c521893f75e0bf9a5e0aad70a6a962b669e7b9d5", # vega
-2124 "e43fae8d5fd1e562607da172369cc0c5ec99b834e42502e682287ff7d12baacc", # vega fp16
-2125 "c6f79f7416a882891957b815fbdfd6edfaa253c43970b1a25ef14e217599c7bc", # flux
-2126 "daf5e09f67ad09a909f58a01298fec0132324634cb8fca2a604c3a240c2c453f", # jaguar mlx q8
-2127 "3f62bfb6bbde05f01435129326166c44aeb113ac0d9f735f31ed3f7dd04f6980", # hidream i1
-2128 "22f866f3c96a92bc61e9965cf366d706db942ad047ba8cb82109edcd4e68fa40", # sd3 turbo
-2129 "f3fa9d7a8f15741621c1fe82f8a1bcc5c601c900d947ac09fba7016615a252a5", # shap-e
-2130 ],
-2131 layer_256=[
-2132 "48daa3d8f939972e69f044533a4312a941971c18c78255f5e555fa26faf664c1",
-2133 "60f5734a74c342be8b0011fc704e718431839790bcfdc7d7004fc39d70f7fec6",
-2134 "6e76e25b4a55dddfa2eecf4b7ab189a8148658a9f6df165c00170f6ce661033c",
-2135 "2d5249df489fec9137cc3a5e9bda499dd9b72a957ddd8e7ad4e99ff3684bad99",
-2136 "3bf085e701713ed3e79775dafea375c3e2a43659ad1ee788b1b393c0aeff9f0e",
-2137 "efb7976800692772e449c81a739339f59394886590ff3f768b0f9ddd87d2a94c",
-2138 "9b0ac8d127c6c457b2eb8c7236f18c4e4ba9e8bbf27130aa8fe854d7c3f7b1e0",
-2139 "24a9ee3d60cdde6c967f08e4b2ec7088fe1bfe308c6896e73caa874860570a5c",
-2140 "5d6d9d0cc7943eb1b8c16862bfd5bee5c3766d0df027ec837e90fac715ac2bd3",
-2141 "68fb122f7d6c3cfbef320341b2af8f5916678e36a69ed36fa8cfcb19e7d5c43d",
-2142 "11807cb2522cfe99240e5ee2bbeb1ccb42cecca2215102ee872567c7773b28b9",
-2143 "50c46cdddbe9f0162278c69b9a1f818519330e3a91b994272e19b5c789670471", # jaguar mlx q8
-2144 "ffe1c4f55e07c2010ace7b9cf35798bb9f431bc954a32784e5acbdc16acc0364", # hidream i1
-2145 "146ea48d234e05a934db9d8988e9a9dd86b2ac70f535eaa550ecb0ee23ec135e", # sd3 turbo
-2146 "d97560cf9704cf71711f6121df2bf55e55a1eda4b574a6ddba074767420bc8c3",
-2147 ],
-2148 )
-2149 )
-2150 repo = "laion/CLIP-ViT-g-14-laion2B-s12B-b42K"
-2151 series, comp = make_mir_tag(repo)
-2152 mir_db.add(
-2153 mir_entry(
-2154 domain="info",
-2155 arch="vit",
-2156 series=series,
-2157 comp=comp,
-2158 repo=repo,
-2159 pkg={0: {"transformers": "CLIPTextModelWithProjection"}},
-2160 identifiers=["31.self_attn.k_proj.weight", "text_model.encoder.layers.22.mlp.fc1.weight", "clip-g"],
-2161 file_256=[
-2162 "ca18e0c67c1ef1e64cac22926266765b60688f692307ecc06283d987c5768134", # seaart furry g
-2163 "ec310df2af79c318e24d20511b601a591ca8cd4f1fce1d8dff822a356bcdb1f4", # modelspec sai
-2164 "fa5b2e6f4c2efc2d82e4b8312faec1a5540eabfc6415126c9a05c8436a530ef4", # playground 2.5
-2165 "b84f413eebecbd049b72874c1df533a516510cb5a2489ae58c7e320209cf0ebe", # ssd1b
-2166 "d3df577f6e3799c8e1bd9b40e30133710e02e8e25d0ce48cdcc790e7dfe12d6d", # ssd1b fp16
-2167 "943a2924ee888295a156dd47089d67181d633b782337890af11ef4b15af17ec5", # vega
-2168 "5b98e4a57a9292eeb819d67e2d2100f66f17db723cde4ecea27a7c3741160d0c", # vega fp16
-2169 "4d6effa7a5e600cabf7528ed7234146a13ead1b2c151211d706b293a060b112a", # hidream i1
-2170 "3a6032f63d37ae02bbc74ccd6a27440578cd71701f96532229d0154f55a8d3ff", # modelspec sai
-2171 "162042ac6556e73f93d4172d4c67532c1cbe4dc7a6a8fa7e44dd2e3d7cbb772b", # onnx
-2172 ],
-2173 layer_b3=[
-2174 "d754db276f2d89d2808abb7086b3b8eccee43ac521c128d21a071f3a631474a8",
-2175 "2eb93685b34719e1d1e0541d8902b0a592d95848f80657e32816cf3b152a0f31",
-2176 "e253a5cf3a6242c58037abd6b378bf0281f278e441f28dff7ca1bcfcd3cd6bd8", # ssd1b
-2177 "16d0eec4e55b0aa63cdca4e4d36f78f66a4b1b9605ce3b1089305026f853c3d2", # ssd1b fp16
-2178 "f606463295ecf3bae8920d3d45bb9d180793418b3d08c3e84d4c4135c7dc2aa5", # vega
-2179 "7060993a5eb32d94d1ea8aef7a7301e7be73b199c639c63f8f7cfbfcd2abf10e", # vega fp16
-2180 "b92af95334c657371af6051a91374a41b5455907fa6622bb66a8c112dc511600", # hidream i1
-2181 ],
-2182 layer_256=[
-2183 "270e998633eb22145100a3889a62ca270d5080654735e5ff8dda09a7c233af8d",
-2184 "df18800c2a9d9318c4323d991a0fb24a6a9afceb41bea203812f60517c301536",
-2185 "4c228b104f6b9b383e0808c9baa1998957f5125d8f90a4d98c1a86e71edd72dc", # ssd1b
-2186 "f7fc81d8b5ae91ec28a5106ecc0d067be9a94fd3f394c4aa4686ed131ce5a5b3", # ssd1b fp16
-2187 "61ab42bd5c0fcb9fd3db1d4014cb844ccae8dc17fd69a108cf077a573d092946", # vega
-2188 "6c64e36cdda3bec7067e94b05619f882f5d31070792acaadac60ddbef580453a", # vega fp16
-2189 "43c9e64995b485a7f128771c48defce128640df28e65c7f79537d472f43ebe46", # hidream i1
-2190 ],
-2191 )
-2192 )
-2193 repo = "laion/CLIP-ViT-H-14-laion2B-s32B-b79K"
-2194 series, comp = make_mir_tag(repo)
-2195 mir_db.add(
-2196 mir_entry(
-2197 domain="info",
-2198 arch="vit",
-2199 series=series,
-2200 comp=comp,
-2201 repo=repo,
-2202 pkg={
-2203 0: {"transformers": "CLIPModel"},
-2204 },
-2205 file_256=[
-2206 "036e6e2bd49697511f4f8b8cb5ee465f93025f7a69a145eadeb9a881ace9b18d",
-2207 "0084e75319a50ad85ef45377bad5bc38f2f58824459eb690048d51c9f8863be5", # open clip
-2208 "64a7ef761bfccbadbaa3da77366aac4185a6c58fa5de5f589b42a65bcc21f161", # wan sai
-2209 ],
-2210 layer_b3=[
-2211 "227f26ed63120b9034f4a0c90b6b37eede721a8260f2c1e8f7ea3ccc0d109e7e",
-2212 "3a38ffd1b60499cf2f451f3065079ff26efb9190a86f23ad1c8d993bbeb9af05", # open clip
-2213 "ce06cf1fd684269ee96631b2bf9334c6ecde6a84a55760dfa0d9d2a6411f28e4", # wan sai
-2214 ],
-2215 layer_256=[
-2216 "130a94ed12569e099196a6ca27388181922e20148dee5bcb58c5e309acfc2352",
-2217 "cfdbd3fd2b90b64ba12d395a62dd7c3c3ea3e811f0a54593e91bae6516ca5061", # open clip
-2218 "9125ce5970c649d6f9368c25493d3aaa6b41e224d4cc427e955115f7b7e53d1c", # wan sai
-2219 ],
-2220 )
-2221 )
-2222 repo = "zai-org/chatglm3-6b" # formerly THUDM
-2223 series, comp = make_mir_tag(repo)
-2224 mir_db.add(
-2225 mir_entry(
-2226 domain="info",
-2227 arch="aet",
-2228 series=series,
-2229 comp=comp,
-2230 repo=repo,
-2231 pkg={
-2232 0: {"transformers": "AutoModel"},
-2233 },
-2234 file_256=[
-2235 "0054d03310248928fdabdeef3fdc753170218dc49a1e9eb5f98323e27683f654", # kolors
-2236 "b1052386eac358a18add3d0f92521c85ab338979da8eeb08a6499555b857f80d",
-2237 ],
-2238 layer_b3=[
-2239 "a45dfba6a9fa8739777c76deb845fc9589b40f88670d3ce4661646a7b7b1d481", # kolors
-2240 ],
-2241 layer_256=[
-2242 "174924fd7a07f370bb6fcd1ad07a73eecb7de901f15eefb80f420c1042c47d44", # kolors
-2243 ],
-2244 )
-2245 )
-2246 base_arch, base_series, base_comp = tag_base_model(repo_path="Qwen/Qwen2-7B-beta", class_name="Qwen2Model")
-2247 repo = "ByteDance-Seed/BAGEL-7B-MoT"
-2248 series, comp = make_mir_tag(repo)
-2249 mir_db.add(
-2250 mir_entry(
-2251 domain="info",
-2252 arch=base_arch,
-2253 series=base_series,
-2254 comp=series,
-2255 repo=repo,
-2256 pkg={0: {"Bagel": "app"}},
-2257 )
-2258 )
-2259
-2260
-2261def add_mir_audio(mir_db: MIRDatabase):
-2262 """Create MIR audio modality entries"""
-2263 repo = "facebook/audiogen-medium"
-2264 series, comp = make_mir_tag(repo)
-2265 mir_db.add(
-2266 mir_entry(
-2267 domain="info",
-2268 arch="art",
-2269 series=series,
-2270 comp=comp,
-2271 repo=repo,
-2272 pkg={
-2273 0: {
-2274 "audiocraft": "models.AudioGen",
-2275 "generation": {"duration": 5},
-2276 "stage_2": {
-2277 "audiocraft": ".data.audioaudio_write",
-2278 "generation": {"strategy": "loudness", "loudness_compressor": True},
-2279 },
-2280 }
-2281 },
-2282 )
-2283 )
-2284 repo = "parler-tts/parler-tts-tiny-v1"
-2285 series, comp = make_mir_tag(repo)
-2286 mir_db.add(
-2287 mir_entry(
-2288 domain="info",
-2289 arch="art",
-2290 series=series,
-2291 comp=comp,
-2292 repo=repo,
-2293 pkg={
-2294 0: {
-2295 "parler_tts": "ParlerTTSForConditionalGeneration",
-2296 "generation": {"return_tensors": "pt"},
-2297 },
-2298 },
-2299 )
-2300 )
-2301 repo = "Zuellni/snac-24khz-ST"
-2302 series, comp = make_mir_tag(repo)
-2303 (
-2304 mir_db.add(
-2305 mir_entry(
-2306 domain="info",
-2307 arch="gan",
-2308 series=series,
-2309 comp=comp,
-2310 repo=repo,
-2311 pkg={
-2312 0: {
-2313 "snac": "SNAC",
-2314 },
-2315 "1": {
-2316 "mlx_audio": "tts.generate.generate_audio",
-2317 },
-2318 },
-2319 file_256=["e61ae2f638f56ee07a37592cd5a6a9e7d642560ddc78a76ee4a7f96d6922f1be", "973ee1be4032319fd9685ec54eee1b93e79c7bc98c786e67f17c04669714f11d"],
-2320 layer_b3=["18307b00460a64cc4893f9061592ce8d7e15b70fc54065cc8ae0f0155381ec46", "d599b1bb36dee3cee4674b7922fcd69e5ec05b74413f611d21cfdfdf8f9b6119"],
-2321 layer_256=["35ba9aa1feb931010559a178fcac243673d2efdd1396a4b69d406c9853a88300", "5a22c4707ed6c928043f23b59f2d102a579db3a9af41cf6e60d7c3958f182841"],
-2322 )
-2323 ),
-2324 )
-2325 repo = "parler-tts/parler-tts-large-v1"
-2326 series, comp = make_mir_tag(repo)
-2327 mir_db.add(
-2328 mir_entry(
-2329 domain="info",
-2330 arch="art",
-2331 series=series,
-2332 comp=comp,
-2333 repo=repo,
-2334 pkg={
-2335 0: {
-2336 "parler_tts": "ParlerTTSForConditionalGeneration",
-2337 "generation": {"return_tensors": "pt"},
-2338 },
-2339 },
-2340 )
-2341 )
-2342 repo = "hexgrad/Kokoro-82M"
-2343 series, comp = make_mir_tag(repo)
-2344 mir_db.add(
-2345 mir_entry(
-2346 domain="info",
-2347 arch="gan",
-2348 series=series,
-2349 comp=comp,
-2350 repo=repo,
-2351 pkg={
-2352 0: {"kokoro": "KPipeline"},
-2353 1: {
-2354 "mlx_audio": "tts.generate.generate_audio",
-2355 "generation": {"audio_format": "wav", "join_audio": True, "verbose": False},
-2356 },
-2357 },
-2358 file_256=[
-2359 "5a5cb3d87478f2e74dfca208ee52209ccfce024095e137097fd276026506e45f",
-2360 "496dba118d1a58f5f3db2efc88dbdc216e0483fc89fe6e47ee1f2c53f18ad1e4",
-2361 ],
-2362 layer_b3=[
-2363 "3e9b5017cfe67a7804ac717b18b6add42ffc0bd3353490df2bcc520eaaef79b6",
-2364 "379660a87a64524bab69a267e3d9580f04b5eec4f7e3fbd48c6597d164d9b17d", # safetensors
-2365 "997f154f5a78879ef3ba1a1556977c40b28b9c21076b8f583f752c57ecc36e93" # pytorch
-2366 "2dc3dba29452b85ea85266084a6248f9e0efe642d5f75b43e64f25b9f2837f92",
-2367 ],
-2368 layer_256=[
-2369 "dbedf0e2115aa309b92689f86534be4a77b91d7900365e1717879fbb19b849f6",
-2370 "2c68574571b3f9229e015a909788116ea2251142e29c1bd5c687863192124e8b",
-2371 ],
-2372 )
-2373 )
-2374 repo = "freddyaboulton/silero-vad"
-2375 series, comp = make_mir_tag(repo)
-2376 mir_db.add(
-2377 mir_entry(
-2378 domain="info",
-2379 arch="stst",
-2380 series=series,
-2381 comp=comp,
-2382 repo=repo,
-2383 pkg={
-2384 0: {
-2385 "onnx": "onnx",
-2386 },
-2387 1: {
-2388 "mlx_audio": "tts.generate.generate_audio",
-2389 "generation": {"audio_format": "wav", "join_audio": True, "verbose": False},
-2390 },
-2391 },
-2392 file_256=["591f853590d11ddde2f2a54f9e7ccecb2533a8af7716330e8adfa6f3849787a9"],
-2393 layer_b3=[
-2394 "41ca5931452b3ffee588c6c7e5bd327c4e914141604eaf3fd05f4a790ac83bb2",
-2395 "7dc736cd5d840182792bde4edfbf5ddc5aeaf16826a9c72d1ba8166c1e3fab9b",
-2396 "6e2c1bdbad74f56663ffb5710c7cb849a2b91ba331d81acdba47a21f69107434", # onnx
-2397 "ab5ff443aece9171af5e7603d0b4309d3ecc934e3940ccedefff10f0b54b931e", # onnx vad
-2398 # "7939427700c3b4d91428a490bde1a6d893f63ee5d79b86f68de9e89c7094d3e7" # onnx # <- clip-g ?? unet? inaccurate test at layer level
-2399 ],
-2400 layer_256=[
-2401 "2ffef1834d5fe14ad8db58fc78d769d5dc38dda5eddbfc396786f74b326215fd",
-2402 # "94ea015f5f7f65b1d8e80f7d52859535e7761d7ed2752e24d57a8d9d9da96672", # onnx lose reliability with layer search apparently
-2403 ],
-2404 ),
-2405 )
-2406 repo = "facebook/wav2vec2-conformer-rope-large-960h-ft"
-2407 series, comp = make_mir_tag(repo)
-2408 mir_db.add(
-2409 mir_entry(
-2410 domain="info",
-2411 arch="stst",
-2412 series=series,
-2413 comp=comp,
-2414 repo=repo,
-2415 pkg={
-2416 0: {
-2417 "transformers": "Wav2Vec2ConformerForCTC",
-2418 },
-2419 },
-2420 file_256=["97bb9761fb71ec1225100bc81ccf7d002e0d0ba3d0604c1fd2dbda7d7d491f1d"],
-2421 layer_b3=["6c9c5642aa8dce62bcb3eb577bc519619a2d868005c767c5e65371c583a8a8eb"],
-2422 layer_256=["1afcfda68307a75caa1a1c4456cf97e20c7914e8aba828006e9fe17e8675a79d"],
-2423 ),
-2424 )
-2425 repo = "canopylabs/orpheus-3b-0.1-ft"
-2426 series, comp = make_mir_tag(repo)
-2427 mir_db.add(
-2428 mir_entry(
-2429 domain="info",
-2430 arch="art",
-2431 series=series,
-2432 comp=comp,
-2433 repo=repo,
-2434 pkg={
-2435 0: {
-2436 "orpheus_tts": "OrpheusModel",
-2437 "generation": {"max_model_len": 2048},
-2438 },
-2439 1: {
-2440 "mlx_audio": "tts.generate.generate_audio",
-2441 "generation": {"audio_format": "wav", "join_audio": True, "verbose": False},
-2442 },
-2443 },
-2444 )
-2445 )
-2446 repo = "OuteAI/OuteTTS-0.3-1B"
-2447 series, comp = make_mir_tag(repo)
-2448 mir_db.add(
-2449 mir_entry(
-2450 domain="info",
-2451 arch="art",
-2452 series=series,
-2453 comp=comp,
-2454 repo=repo,
-2455 pkg={
-2456 0: {"outetts": "InterfaceHF"},
-2457 1: {
-2458 "mlx_audio": "tts.generate.generate_audio",
-2459 "generation": {"audio_format": "wav", "join_audio": True, "verbose": False},
-2460 },
-2461 },
-2462 )
-2463 )
-2464
-2465
-2466def add_mir_lora(mir_db: MIRDatabase):
-2467 """Create MIR lora entries"""
-2468 mir_db.add(
-2469 mir_entry(
-2470 domain="info",
-2471 arch="lora",
-2472 series="dmd",
-2473 comp=sdxl_series,
-2474 repo="tianweiy/DMD2",
-2475 pkg={
-2476 0: {
-2477 "diffusers": {"load_lora_weights": {}},
-2478 "generation": {"num_inference_steps": 4, "guidance_scale": 0, "timesteps": [999, 749, 499, 249]},
-2479 "scheduler": {"ops.scheduler.lcm": ""},
-2480 }
-2481 },
-2482 file_256=[
-2483 "b3d9173815a4b595991c3a7a0e0e63ad821080f314a0b2a3cc31ecd7fcf2cbb8",
-2484 "a374289e9446d7f14d2037c4b3770756b7b52c292142a691377c3c755010a1bb",
-2485 ],
-2486 )
-2487 )
-2488 mir_db.add(
-2489 mir_entry(
-2490 domain="info",
-2491 arch="lora",
-2492 series="dpo",
-2493 comp=sdxl_series,
-2494 repo="radames/sdxl-DPO-LoRA",
-2495 pkg={
-2496 0: {
-2497 "diffusers": {"load_lora_weights": {}},
-2498 "generation": {"guidance_scale": 7.5, "num_inference_steps": 4},
-2499 "scheduler": {"ops.scheduler.dpm": {"algorithm_type": "sde-dpmsolver++", "use_karras_sigmas": True, "order": 2}},
-2500 },
-2501 },
-2502 file_256=[
-2503 "666f71a833fc41229ec7e8a264fb7b0fcb8bf47a80e366ae7486c18f38ec9fc0",
-2504 "6b1dcbfb234d7b6000948b5b95ccebc8f903450ce2ba1b50bc3456987c9087ad",
-2505 ],
-2506 )
-2507 )
-2508 mir_db.add(
-2509 mir_entry(
-2510 domain="info",
-2511 arch="lora",
-2512 series="flash",
-2513 comp=sdxl_series,
-2514 repo="jasperai/flash-sdxl",
-2515 pkg={
-2516 0: {
-2517 "diffusers": {"load_lora_weights": {}},
-2518 "scheduler": "ops.scheduler.lcm",
-2519 }
-2520 },
-2521 file_256=["afe2ca6e27c4c6087f50ef42772c45d7b0efbc471b76e422492403f9cae724d7"],
-2522 ),
-2523 )
-2524 mir_db.add(
-2525 mir_entry(
-2526 domain="info",
-2527 arch="lora",
-2528 series="flash",
-2529 comp="pixart-alpha",
-2530 repo="jasperai/flash-pixart",
-2531 pkg={
-2532 0: {"diffusers": {"load_lora_weights": {}}},
-2533 },
-2534 file_256=["99ef037fe3c1fb6d6bbefdbb85ad60df434fcc0577d34c768d752d60cf69681b"],
-2535 )
-2536 )
-2537 mir_db.add(
-2538 mir_entry(
-2539 domain="info",
-2540 arch="lora",
-2541 series="flash",
-2542 comp=sd3_series,
-2543 repo="jasperai/flash-sd3",
-2544 pkg={
-2545 0: {"diffusers": {"load_lora_weights": {}}},
-2546 },
-2547 file_256=["85fce13c36e3739aa42930f745eb9fceb6c53d53fb17e2a687e3234c1a58ee15"],
-2548 )
-2549 )
-2550 mir_db.add(
-2551 mir_entry(
-2552 domain="info",
-2553 arch="lora",
-2554 series="flash",
-2555 comp=sd1_series,
-2556 repo="jasperai/flash-sd",
-2557 pkg={
-2558 0: {"diffusers": {"load_lora_weights": {}}, "generation": {"num_inference_steps": 4, "guidance_scale": 0}},
-2559 },
-2560 file_256=["99353444c1a0f40719a1b3037049dbd24800317979a73c312025c05af3574a5f"],
-2561 ),
-2562 )
-2563 mir_db.add(
-2564 mir_entry(
-2565 domain="info",
-2566 arch="lora",
-2567 series="hyper",
-2568 comp=sdxl_series,
-2569 repo="ByteDance/Hyper-SD",
-2570 pkg={0: {"diffusers": {"load_lora_weights": {"fuse": 1.0}}}},
-2571 file_256={
-2572 "0b97f447b5878323a28fbe7c51ba7acebd21f4d77552ba77b04b11c8911825b6": {"num_inference_steps": 12},
-2573 "55b51334c85061afff5eff7c550b61963c8b8607a5868bbe4f26db49374719b1": {"num_inference_steps": 8},
-2574 "c912df184c5116792d2c604d26c6bc2aa916685f4a793755255cda1c43a3c78a": {"num_inference_steps": 1, "guidance_scale": 0.0},
-2575 "69b25c0187ced301c3603c599c0bc509ac99b8ac34db89a2aecc3d5f77a35187": {"num_inference_steps": 2, "guidance_scale": 0.0},
-2576 "12f81a27d00a751a40d68fd15597091896c5a90f3bd632fb6c475607cbdad76e": {"num_inference_steps": 4, "guidance_scale": 0.0},
-2577 "ca689190e8c46038550384b5675488526cfe5a40d35f82b27acb75c100f417c1": {"num_inference_steps": 8, "guidance_scale": 0.0},
-2578 },
-2579 ),
-2580 )
-2581 mir_db.add(
-2582 mir_entry(
-2583 domain="info",
-2584 arch="lora",
-2585 series="hyper",
-2586 comp=dev_series,
-2587 repo="ByteDance/Hyper-SD",
-2588 pkg={0: {"diffusers": {"load_lora_weights": {"fuse": 0.125}}}},
-2589 file_256={
-2590 "6461f67dfc1a967ae60344c3b3f350877149ccab758c273cc37f5e8a87b5842e": {"num_inference_steps": 16, "guidance_scale": 0.0},
-2591 "e0ab0fdf569cd01a382f19bd87681f628879dea7ad51fe5a3799b6c18c7b2d03": {"num_inference_steps": 8, "guidance_scale": 0.0},
-2592 },
-2593 ),
-2594 )
-2595 mir_db.add(
-2596 mir_entry(
-2597 domain="info",
-2598 arch="lora",
-2599 series="hyper",
-2600 comp=sd3_series,
-2601 repo="ByteDance/Hyper-SD",
-2602 pkg={0: {"diffusers": {"load_lora_weights": {"fuse": 0.125}}}},
-2603 file_256={
-2604 "5b4d0b99d58deb811bdbbe521a06f4dbf56a2e9148ff3211c594e0502b656bc9": {"num_inference_steps": 16},
-2605 "0ee4e529abd17b06d4295e3bb91c0d4ddae393afad86b2b43c4f5eeb9e401602": {"num_inference_steps": 4},
-2606 "fc6a3e73e14ed11e21e4820e960d7befcffe7e333850ada9545f239e9aa6027e": {"num_inference_steps": 8},
-2607 },
-2608 ),
-2609 )
-2610 mir_db.add(
-2611 mir_entry(
-2612 domain="info",
-2613 arch="lora",
-2614 series="hyper",
-2615 comp=sd1_series,
-2616 repo="ByteDance/Hyper-SD",
-2617 pkg={0: {"diffusers": {"load_lora_weights": {}}}},
-2618 file_256={
-2619 "64b98437383537cd968fda6f87a05c33160ece9c79ff4757949a1e212ff78361": {"num_inference_steps": 12},
-2620 "f6123d5b950d5250ab6c33600e27f4dcf71b3099ebf888685e01e9e8117ce482": {"num_inference_steps": 8},
-2621 "a04fd9a535c1e56d38f7590ee72a13fd5ca0409853b4fff021e5a9482cf1ca3b": {"num_inference_steps": 1, "guidance_scale": 0.0},
-2622 "2f26dcc1d883feb07557a552315baae2ca2a04ac08556b08a355a244547e8c3a": {"num_inference_steps": 2, "guidance_scale": 0.0},
-2623 "c5dd058616461ed5053e2b14eec4dbe3fa0eea3b13688642f6d6c80ea2ba5958": {"num_inference_steps": 4, "guidance_scale": 0.0},
-2624 "91fc3186236e956d64dbb4357f2e120c69b968b78af7d2db9884a5ca74d3cd13": {"num_inference_steps": 8, "guidance_scale": 0.0},
-2625 },
-2626 )
-2627 )
-2628 mir_db.add(
-2629 mir_entry(
-2630 domain="info",
-2631 arch="lora",
-2632 series="lcm",
-2633 comp=sdxl_series,
-2634 repo="latent-consistency/lcm-lora-sdxl",
-2635 pkg={
-2636 0: {
-2637 "diffusers": {"load_lora_weights": {"fuse": 1.0}},
-2638 "scheduler": {"ops.scheduler.lcm": {"timestep_spacing": "trailing"}},
-2639 "generation": {"num_inference_steps": 8},
-2640 },
-2641 },
-2642 file_256=["a764e6859b6e04047cd761c08ff0cee96413a8e004c9f07707530cd776b19141"],
-2643 )
-2644 )
-2645 mir_db.add(
-2646 mir_entry(
-2647 domain="info",
-2648 arch="lora",
-2649 series="lcm",
-2650 comp=ssd_series,
-2651 repo="latent-consistency/lcm-lora-ssd-1b",
-2652 pkg={0: {"diffusers": {"load_lora_weights": {}}, "generation": {"num_inference_steps": 8}}},
-2653 file_256=["7adaaa69db6f011058a19fd1d5315fdf19ef79fcd513cdab30e173833fd5c59b"],
-2654 ),
-2655 )
-2656 mir_db.add(
-2657 mir_entry(
-2658 domain="info",
-2659 arch="lora",
-2660 series="lcm",
-2661 comp=vega_series,
-2662 repo="segmind/Segmind-VegaRT",
-2663 pkg={0: {"diffusers": {"load_lora_weights": {}}, "gen_kwargs": {"num_inference_steps": 8}}},
-2664 file_256=["9b6e8cd833fa205eaeeed391ca623a6f2546e447470bd1c5dcce3fa8d2f26afb"],
-2665 ),
-2666 )
-2667 mir_db.add(
-2668 mir_entry(
-2669 domain="info",
-2670 arch="lora",
-2671 series="lcm",
-2672 comp=sd1_series,
-2673 repo="latent-consistency/lcm-lora-sdv1-5",
-2674 pkg={0: {"diffusers": {"load_lora_weights": {}}, "generation": {"num_inference_steps": 8}}},
-2675 file_256=["8f90d840e075ff588a58e22c6586e2ae9a6f7922996ee6649a7f01072333afe4"],
-2676 ),
-2677 )
-2678 mir_db.add(
-2679 mir_entry(
-2680 domain="info",
-2681 arch="lora",
-2682 series="lightning",
-2683 comp=sdxl_series,
-2684 repo="ByteDance/SDXL-Lightning",
-2685 pkg={0: {"diffusers": {"load_lora_weights": {}}, "generation": {"num_inference_steps": 4, "guidance_scale": 0}}},
-2686 ),
-2687 )
-2688 mir_db.add(
-2689 mir_entry(
-2690 domain="info",
-2691 arch="lora",
-2692 series="pcm",
-2693 comp=sdxl_series,
-2694 repo="wangfuyun/PCM_Weights",
-2695 pkg={0: {"diffusers": {"load_lora_weights": {}}}},
-2696 file_256={
-2697 "0365f6107250a4fed1b83e8ae6a070065e026a2ba54bff65f55a50284232bbe6": {"num_inference_steps": 4, "guidance_scale": 0.0},
-2698 "04ea827435d5750e63d113dc509174b4f6e8a069ff8f91970c3d25299c10b1f8": {"num_inference_steps": 16},
-2699 "7eb353b2abcaabab6251ba4e17d6cbe2e763feb0674b0f950555552212b44621": {"num_inference_steps": 16},
-2700 "a85cf70ac16ed42011630a5cd6b5927722cb7c40a2107eff85e2670f9a38c893": {"num_inference_steps": 4}, # float16
-2701 "9f7f13bb019925eacd89aeff678e4fd831f7b60245b986855dff6634aee4eba9": {"num_inference_steps": 4},
-2702 "3b9c970a3e4c0e182931e71b3f769c1956f16c6b06db98b4d67236790d4d0b1d": {"num_inference_steps": 8},
-2703 "7f04ba8911b4c25ef2c7cbf74abcb6daa3b4f0e4bc6a03896bdae7601f2f180b": {"num_inference_steps": 8},
-2704 "13fb038025ce9dad93b8ee1b67fc81bac8affb59a77b67d408d286e0b0365a1d": {"num_inference_steps": 16, "guidance_scale": 0.0},
-2705 "3442eff271aa3b60a094fd6f9169d03e49e4051044a974f6fcf690507959191f": {"num_inference_steps": 16, "guidance_scale": 0.0},
-2706 "242cbe4695fe3f2e248faa71cf53f2ccbf248a316973e4b2f38ab9e34f35a5ab": {"num_inference_steps": 2, "guidance_scale": 0.0},
-2707 "e1f600491bb8e0cd94f41144321e44fdb2cb346447f31e71f6e53f1c24cccfbf": {"num_inference_steps": 2, "guidance_scale": 0.0},
-2708 "d0bf40a7f280829195563486bec7253f043a06b1f218602b20901c367641023e": {"num_inference_steps": 4, "guidance_scale": 0.0},
-2709 "212150d7953627fb89df99aad579d6763645a1cb2ef26b19fee8b398d5e5ff4d": {"num_inference_steps": 4, "guidance_scale": 0.0},
-2710 "e80fcf46d15f4d3821d3d9611bdb3022a4a8b647b2536833b168d317a91e4f74": {"num_inference_steps": 8, "guidance_scale": 0.0},
-2711 "56ed9dc9f51f4bb0d6172e13b7947f215c347fc0da341c8951b2c12b9507d09e": {"num_inference_steps": 8, "guidance_scale": 0.0},
-2712 },
-2713 )
-2714 )
-2715 mir_db.add(
-2716 mir_entry(
-2717 domain="info",
-2718 arch="lora",
-2719 series="pcm",
-2720 comp=sd1_series,
-2721 repo="wangfuyun/PCM_Weights",
-2722 pkg={0: {"diffusers": {"load_lora_weights": {}}}},
-2723 file_256={
-2724 "b80b27dd6504f1c3a7637237dda86bc7e26fa5766da30c4fc853c0a1d46bad31": {"num_inference_steps": 4, "guidance_scale": 0.0},
-2725 "8f605ffde3616592deb37ed8c6bacb83fe98963c1fd0883c2a4f93787098aa45": {"num_inference_steps": 16},
-2726 "fa6acb94f11dba3bf4120af5a12e3c88cd2b9572d43ec1a6fb04eede9f32829e": {"num_inference_steps": 4},
-2727 "bff3d4499718b61455b0757b5f8d98fe23e73a768b538c82ecf91c693b69dbcd": {"num_inference_steps": 8},
-2728 "c7ac2fa3df3a5b7080ebe63f259ab13630014f104c93c3c706d77b05cc48506b": {"num_inference_steps": 16, "guidance_scale": 0.0},
-2729 "4c5f27a727d12146de4b1d987cee3343bca89b085d12b03c45297af05ce88ef4": {"num_inference_steps": 2, "guidance_scale": 0.0},
-2730 "29278bc86274fdfc840961e3c250758ff5e2dc4666d940f103e78630d5b879d3": {"num_inference_steps": 4, "guidance_scale": 0.0},
-2731 "41a7f0b966d18f643d16c4401f0b5ef6b9ef7362c20e17128322f17874709107": {"num_inference_steps": 8, "guidance_scale": 0.0},
-2732 },
-2733 )
-2734 )
-2735 mir_db.add(
-2736 mir_entry(
-2737 domain="info",
-2738 arch="lora",
-2739 series="pcm",
-2740 comp=sd3_series,
-2741 repo="wangfuyun/PCM_Weights",
-2742 pkg={0: {"diffusers": {"load_lora_weights": {}}}},
-2743 file_256={
-2744 "8a45878ecc34e53855fe21146cb6ef32682053b7c4eacc013be89fb08c4c19d8": {"num_inference_steps": 2, "guidance_scale": 1.2},
-2745 "9444a5cead551c56c4d1c455ce829ba9f96f01fbcca31294277e0862a6a15b76": {"num_inference_steps": 4, "guidance_scale": 1.2},
-2746 "e365902c208cbc0456ca5e7c41a490f637c15f3f7b98691cbba21f96a8c960b4": {"num_inference_steps": 4, "guidance_scale": 1.2},
-2747 "3550fa018cd0b60d9e36ac94c31b30f27e402d3855ed63e47668bb181b35a0ad": {"num_inference_steps": 4, "guidance_scale": 1.2},
-2748 },
-2749 )
-2750 )
-2751 mir_db.add(
-2752 mir_entry(
-2753 domain="info",
-2754 arch="lora",
-2755 series="slam",
-2756 comp=sdxl_series,
-2757 repo="alimama-creative/slam-lora-sdxl",
-2758 pkg={
-2759 0: {
-2760 "diffusers": {"load_lora_weights": {}},
-2761 "scheduler": {"ops.scheduler.lcm": {"timestep_spacing": "trailing"}},
-2762 "generation": {"num_inference_steps": 4, "guidance_scale": 1},
-2763 }
-2764 },
-2765 file_256=["22569a946b0db645aa3b8eb782c674c8e726a7cc0d655887c21fecf6dfe6ad91"],
-2766 )
-2767 )
-2768 mir_db.add(
-2769 mir_entry(
-2770 domain="info",
-2771 arch="lora",
-2772 series="slam",
-2773 comp=sd1_series,
-2774 repo="alimama-creative/slam-sd1.5",
-2775 pkg={0: {"diffusers": {"load_lora_weights": {}}}},
-2776 )
-2777 )
-2778 mir_db.add(
-2779 mir_entry(
-2780 domain="info",
-2781 arch="lora",
-2782 series="spo",
-2783 comp=sdxl_series,
-2784 repo="SPO-Diffusion-Models/SPO-SDXL_4k-p_10ep_LoRA",
-2785 pkg={0: {"diffusers": {"load_lora_weights": {}}, "generation": {"guidance_scale": 5.0}}},
-2786 file_256=["0b9896f30d29daa5eedcfc9e7ad03304df6efc5114508f6ca9c328c0b4f057df"],
-2787 ),
-2788 )
-2789 mir_db.add(
-2790 mir_entry(
-2791 domain="info",
-2792 arch="lora",
-2793 series="spo",
-2794 comp=sd1_series,
-2795 repo="SPO-Diffusion-Models/SPO-SD-v1-5_4k-p_10ep_LoRA",
-2796 pkg={0: {"diffusers": {"load_lora_weights": {}}, "generation": {"guidance_scale": 7.5}}},
-2797 file_256=["1be130c5be2de0beacadd3bf0bafe3bedd7e7a380729932a1e369fb29efa86f4"],
-2798 ),
-2799 )
-2800 mir_db.add(
-2801 mir_entry(
-2802 domain="info",
-2803 arch="lora",
-2804 series="tcd",
-2805 comp=sdxl_series,
-2806 repo="h1t/TCD-SDXL-LoRA",
-2807 pkg={
-2808 0: {
-2809 "diffusers": {"load_lora_weights": {}},
-2810 "generation": {"num_inference_steps": 4, "guidance_scale": 0, "eta": 0.3},
-2811 "scheduler": {"ops.scheduler.tcd": {}},
-2812 }
-2813 },
-2814 file_256=["2c777bc60abf41d3eb0fe405d23d73c280a020eea5adf97a82a141592c33feba"],
-2815 ),
-2816 )
-2817 mir_db.add(
-2818 mir_entry(
-2819 domain="info",
-2820 arch="lora",
-2821 series="tcd",
-2822 comp=sd1_series,
-2823 repo="h1t/TCD-SD15-LoRA",
-2824 pkg={0: {"diffusers": {"load_lora_weights": {}}}},
-2825 file_256=["eaecb24a1cda4411eab67275b1d991071216ac93693e8fa0c9226c9df0386232"],
-2826 layer_b3=["90158259812a89beb8874216009c799f420334aac49bbf4fa1bf0ebf4bbd256b"],
-2827 layer_256=["e9825b81bca684126ac3cc8867d2ebc655f74268bc26bea4e4b7e58a52ad6c75"],
-2828 )
-2829 )
-2830 mir_db.add(
-2831 mir_entry(
-2832 domain="info",
-2833 arch="lora",
-2834 series="turbo",
-2835 comp=sdxl_series,
-2836 file_256=["a599c42a9f4f7494c7f410dbc0fd432cf0242720509e9d52fa41aac7a88d1b69"],
-2837 )
-2838 )
-2839 mir_db.add(
-2840 mir_entry(
-2841 domain="info",
-2842 arch="lora",
-2843 series="turbo",
-2844 comp=dev_series,
-2845 repo="alimama-creative/FLUX.1-Turbo-Alpha",
-2846 pkg={
-2847 0: {
-2848 "diffusers": {"load_lora_weights": {"fuse": 0.125}},
-2849 "generation": {"guidance_scale": 3.5, "num_inference_steps": 8, "max_sequence_length": 512},
-2850 }
-2851 },
-2852 file_256=["77f7523a5e9c3da6cfc730c6b07461129fa52997ea06168e9ed5312228aa0bff"],
-2853 )
-2854 )
-2855 mir_db.add(
-2856 mir_entry(
-2857 domain="info",
-2858 arch="lora",
-2859 series="turbo",
-2860 comp=sd3_series,
-2861 repo="tensorart/stable-diffusion-3.5-medium-turbo",
-2862 pkg={0: {"diffusers": {"load_lora_weights": {"fuse": 1.0}}, "scheduler": {"ops.scheduler.flow-match": {"shift": 5}}}},
-2863 file_256={"bdcbdfa3ec8ed838b77b1020eea3bc7917a2d42573688a034feb921fde8b1858": {"num_inference_steps": "4"}},
-2864 )
-2865 )
-2866 mir_db.add(
-2867 mir_entry(
-2868 domain="info",
-2869 arch="lora",
-2870 series="turbo",
-2871 comp=sd3_series,
-2872 repo="tensorart/stable-diffusion-3.5-large-TurboX",
-2873 pkg={0: {"diffusers": {"load_lora_weights": {"fuse": 1.0}}, "scheduler": {"ops.scheduler.flow-match": {"shift": 5}}}},
-2874 file_256={"fae59d1b749c0d14a8fd4c68cc94eaac92876cee7b91fa75cf8fde3160e09548": {"num_inference_steps": "8"}},
-2875 )
-2876 )
-2877
-2878
-2879def add_mir_vae(mir_db: MIRDatabase):
-2880 """Create MIR VAE missing from the database"""
-2881 mir_db.add(
-2882 mir_entry(
-2883 domain="info",
-2884 arch="vae",
-2885 series="tae",
-2886 comp=sd3_series,
-2887 repo="madebyollin/taesd3",
-2888 pkg={0: {"diffusers": "AutoencoderTiny"}},
-2889 file_256=["6f79c1397cb9ce1dac363722dbe70147aee0ccca75e28338f8482fe515891399"],
-2890 )
-2891 )
-2892 mir_db.add(
-2893 mir_entry(
-2894 domain="info",
-2895 arch="vae",
-2896 series="tae",
-2897 comp=sdxl_series,
-2898 repo="madebyollin/taesdxl",
-2899 pkg={0: {"diffusers": "AutoencoderTiny"}},
-2900 file_256=["ff4824aca94dd6111e0340fa749347fb74101060d9712cb5ef1ca8f1cf17502f"],
-2901 )
-2902 )
-2903 mir_db.add(
-2904 mir_entry(
-2905 domain="info",
-2906 arch="vae",
-2907 series="tae",
-2908 comp=sd1_series,
-2909 repo="madebyollin/taesd",
-2910 pkg={0: {"diffusers": "AutoencoderTiny"}},
-2911 file_256=["db169d69145ec4ff064e49d99c95fa05d3eb04ee453de35824a6d0f325513549"],
-2912 )
-2913 )
-2914 mir_db.add(
-2915 mir_entry(
-2916 domain="info",
-2917 arch="vae",
-2918 series="tae",
-2919 comp=dev_series,
-2920 repo="madebyollin/taef1",
-2921 pkg={0: {"diffusers": "AutoencoderTiny"}},
-2922 file_256=["927f7de7f11bbd3b2d5ce402e608d97a7649e0921a9601995b044e8efc81e449"],
-2923 )
-2924 )
-2925 series, comp = make_mir_tag("Qwen/Qwen-Image")
-2926 mir_db.add(
-2927 mir_entry(
-2928 domain="info",
-2929 arch="vae",
-2930 series="kl",
-2931 comp=series,
-2932 # no repo here, may conflict
-2933 pkg={
-2934 0: {"diffusers": "AutoencoderKLQwenImage"},
-2935 },
-2936 file_256=[
-2937 "0c8bc8b758c649abef9ea407b95408389a3b2f610d0d10fcb054fe171d0a8344", # diffusers
-2938 ],
-2939 layer_b3=[
-2940 "64af8fb08d2054c81ad2aef94965be8fb1366fcc6136cb9222ae046550af014b", # diffusers
-2941 ],
-2942 layer_256=[
-2943 "42f255440ef1d379a8a731456bc44312a73a8568716caa6100803990cd5ea7dc", # diffusers
-2944 ],
-2945 )
-2946 )
-2947 series, comp = make_mir_tag("Wan-AI/Wan2.1-I2V-14B-480P-Diffusers")
-2948 sr_series_text2v, _ = make_mir_tag("Skywork/SkyReels-V2-T2V-14B-720P-Diffusers")
-2949 sr_series_image2v, _ = make_mir_tag("Skywork/SkyReels-V2-I2V-14B-720P-Diffusers")
-2950 mir_db.add(
-2951 mir_entry(
-2952 domain="info",
-2953 arch="vae",
-2954 series="wan",
-2955 comp=series,
-2956 # no repo here, may conflict
-2957 pkg={
-2958 0: {
-2959 "diffusers": "AutoencoderKLWan",
-2960 "precision": "ops.precision.float.F32",
-2961 }
-2962 },
-2963 file_256=[
-2964 "d6e524b3fffede1787a74e81b30976dce5400c4439ba64222168e607ed19e793", # diffusers
-2965 "2fc39d31359a4b0a64f55876d8ff7fa8d780956ae2cb13463b0223e15148976b", # sai
-2966 ],
-2967 layer_b3=[
-2968 "f867543d636029ebfc05b8075e572be0b313a83b0470e56bcf4bbad07a6db010", # diffusers
-2969 "6b5b229727a2d4e37993687c62c94ff8519a371ab4103c699ff1f5969ca0b433", # sai
-2970 ],
-2971 layer_256=[
-2972 "121b3974b39263dcca9d644d1b5c9b9251a911b6a8a8e307fcb21ca778e78ed2",
-2973 "364be43a8959012d798d3f98e17d8b5c4b99ba1e70077008dd19acca3ced395e",
-2974 ],
-2975 )
-2976 )
-2977 mir_db.add(
-2978 mir_entry(
-2979 domain="info",
-2980 arch="vae",
-2981 series="wan",
-2982 comp=sr_series_text2v,
-2983 # no repo here, may conflict
-2984 file_256=[],
-2985 layer_b3=[],
-2986 layer_256=[],
-2987 )
-2988 )
-2989 mir_db.add(
-2990 mir_entry(
-2991 domain="info",
-2992 arch="vae",
-2993 series="wan",
-2994 comp=sr_series_image2v,
-2995 # no repo here, may conflict
-2996 file_256=[],
-2997 layer_b3=[],
-2998 layer_256=[],
-2999 )
-3000 )
-3001 series, comp = make_mir_tag("Lightricks/LTX-Video")
-3002 mir_db.add(
-3003 mir_entry(
-3004 domain="info",
-3005 arch="vae",
-3006 series="kl",
-3007 comp=series,
-3008 # no repo here, may conflict
-3009 pkg={
-3010 0: {"diffusers": "AutoencoderKLLTXVideo"},
-3011 },
-3012 file_256=[],
-3013 layer_b3=[],
-3014 layer_256=[],
-3015 )
-3016 )
-3017 series, comp = make_mir_tag("rhymes-ai/Allegro")
-3018 mir_db.add(
-3019 mir_entry(
-3020 domain="info",
-3021 arch="vae",
-3022 series="kl",
-3023 comp=series,
-3024 # no repo here, may conflict
-3025 pkg={
-3026 0: {"diffusers": "AutoencoderKLAllegro"},
-3027 },
-3028 file_256=[],
-3029 layer_b3=[],
-3030 layer_256=[],
-3031 )
-3032 )
-3033 series, comp = make_mir_tag("zai-org/CogVideoX-5b-I2V")
-3034 series_fun, _ = make_mir_tag("alibaba-pai/CogVideoX-Fun-V1.1-5b-Pose")
-3035 series_wish, _ = make_mir_tag("BestWishYsh/ConsisID-preview")
-3036 mir_db.add(
-3037 mir_entry(
-3038 domain="info",
-3039 arch="vae",
-3040 series="cogvideox",
-3041 comp=series,
-3042 # no repo here, may conflict
-3043 pkg={
-3044 0: {"diffusers": "AutoencoderKLCogVideoX"},
-3045 },
-3046 file_256=["a410e48d988c8224cef392b68db0654485cfd41f345f4a3a81d3e6b765bb995e"],
-3047 layer_b3=["246addb8dc798240638bffee4546a3c5c83572139b4a2a602d68b4c4146226eb"],
-3048 layer_256=["43c7e9cb4364e55fd563817f01484ede8a09ff19a8e69eb61a32a12f93d6f66e"],
-3049 )
-3050 )
-3051 mir_db.add(
-3052 mir_entry(
-3053 domain="info",
-3054 arch="vae",
-3055 series="cogvideox",
-3056 comp=series_fun,
-3057 # no repo here, may conflict
-3058 file_256=[],
-3059 layer_b3=[],
-3060 layer_256=[],
-3061 )
-3062 )
-3063 mir_db.add(
-3064 mir_entry(
-3065 domain="info",
-3066 arch="vae",
-3067 series="cogvideox",
-3068 comp=series_wish,
-3069 # no repo here, may conflict
-3070 file_256=[],
-3071 layer_b3=[],
-3072 layer_256=[],
-3073 )
-3074 )
-3075 series, comp = make_mir_tag("nvidia/Cosmos-1.0-Diffusion-7B-Video2World")
-3076 mir_db.add(
-3077 mir_entry(
-3078 domain="info",
-3079 arch="vae",
-3080 series="kl",
-3081 comp=series,
-3082 # no repo here, may conflict
-3083 pkg={
-3084 0: {"diffusers": "AutoencoderKLCosmos"},
-3085 },
-3086 file_256=[],
-3087 layer_b3=[],
-3088 layer_256=[],
-3089 )
-3090 )
-3091 series, comp = make_mir_tag("alibaba-pai/EasyAnimateV5.1-7b-zh-diffusers")
-3092 mir_db.add(
-3093 mir_entry(
-3094 domain="info",
-3095 arch="vae",
-3096 series="kl",
-3097 comp=series,
-3098 # no repo here, may conflict
-3099 pkg={
-3100 0: {"diffusers": "AutoencoderKLMagvit"},
-3101 },
-3102 file_256=[],
-3103 layer_b3=[],
-3104 layer_256=[],
-3105 )
-3106 )
-3107 series, comp = make_mir_tag("hunyuanvideo-community/HunyuanVideo-I2V")
-3108 mir_db.add(
-3109 mir_entry(
-3110 domain="info",
-3111 arch="vae",
-3112 series="kl",
-3113 comp=series,
-3114 # no repo here, may conflict
-3115 pkg={
-3116 0: {"diffusers": "AutoencoderKLHunyuanVideo"},
-3117 },
-3118 file_256=[
-3119 "95d1fc707c1421ccd88ea542838ab4c5d45a5babb48205bac9ce0985525f9818", # pt,
-3120 "7c68a6295f9034a88225fbafb1f3258291a08d57a1fdb938233fa57b1b8f4883",
-3121 "fbe5ea338431bc8ba20f7019b474e83379fe5763abfd562adcc04b1c0d35c728",
-3122 "019973c147e0c3462629d8d06bdbdbb83408f3ebd4ea4b4ae21a99c3cdcb54c0",
-3123 ],
-3124 # layer_b3=[],
-3125 # layer_256=[],
-3126 )
-3127 )
-3128 series, comp = make_mir_tag("genmo/mochi-1-preview")
-3129 mir_db.add(
-3130 mir_entry(
-3131 domain="info",
-3132 arch="vae",
-3133 series="kl",
-3134 comp=series,
-3135 # no repo here, may conflict
-3136 pkg={
-3137 0: {"diffusers": "AutoencoderKLMochi"},
-3138 },
-3139 file_256=[],
-3140 layer_b3=[],
-3141 layer_256=[],
-3142 )
-3143 )
-3144 series, comp = make_mir_tag("rhymes-ai/Allegro")
-3145 mir_db.add(
-3146 mir_entry(
-3147 domain="info",
-3148 arch="vae",
-3149 series="kl",
-3150 comp=series,
-3151 # no repo here, may conflict
-3152 pkg={
-3153 0: {
-3154 "diffusers": "AutoencoderKLAllegro",
-3155 },
-3156 },
-3157 file_256=["47871a698b18f92f15019d361a81cbc8af4676f8eef9a47fd2b95354a39f831a"],
-3158 layer_b3=["93654cbab7541504d2377c66e72943c7fd9947fca2eb1be01bcc8877c322c1e0"],
-3159 layer_256=["bfd496586118165a13243997101fc7cdd4f855b2d8a73ee2b771a4484c4c2f9f"],
-3160 )
-3161 )
-3162 series, comp = make_mir_tag("cvssp/audioldm-s-full-v2")
-3163 mir_db.add(
-3164 mir_entry(
-3165 domain="info",
-3166 arch="vae",
-3167 series="kl",
-3168 comp=series,
-3169 # no repo here, may conflict
-3170 pkg={
-3171 0: {
-3172 "diffusers": "AutoencoderKL",
-3173 },
-3174 },
-3175 file_256=["42f64f7565b23eabde68c9694e39f18b8bba5f7a14f477e7ed4b51e0ea7de8a5"],
-3176 layer_b3=["00959677dae940b9cfdbe5380c8cbb5a6b4951864cd26f8211d74a3d22b4f3de"],
-3177 layer_256=["54d075953d5253a3abac651de070736c1d5510b857a8ab24c624304f428146b6"],
-3178 )
-3179 )
-3180
-3181 series, comp = make_mir_tag("Efficient-Large-Model/Sana_1600M_1024px_BF16_diffusers")
-3182 mir_db.add(
-3183 mir_entry(
-3184 domain="info",
-3185 arch="vae",
-3186 series="dc",
-3187 comp=series,
-3188 # no repo here, may conflict
-3189 pkg={
-3190 0: {"diffusers": "AutoencoderDC"},
-3191 },
-3192 file_256=["15a4b09e56d95b768a0ec9da50b702e21d920333fc9b3480d66bb5c7fad9d87f"],
-3193 layer_b3=["cf4ecc6697d18b0663e4eac58203f1dd6d9fb689cf99adfeadbc0019de0c73d0"],
-3194 layer_256=["abfc39d1a6d71f03dde7bc40fec4a90478a97d17ae1688be9aad00e0512b9bde"],
-3195 )
-3196 )
-3197 series, comp = make_mir_tag("stabilityai/stable-audio-open-1.0")
-3198 mir_db.add(
-3199 mir_entry(
-3200 domain="info",
-3201 arch="vae",
-3202 series="oobleck",
-3203 comp=series,
-3204 # no repo here, may conflict
-3205 pkg={
-3206 0: {"diffusers": "AutoencoderOobleck"},
-3207 },
-3208 # file_256=[],
-3209 # layer_b3=[],
-3210 # layer_256=[],
-3211 )
-3212 )
-3213 series, comp = make_mir_tag("stable-video-diffusion-img2vid-xt")
-3214 mir_db.add(
-3215 mir_entry(
-3216 domain="info",
-3217 arch="vae",
-3218 series="kl",
-3219 comp=series,
-3220 # no repo here, may conflict
-3221 pkg={
-3222 0: {"diffusers": "AutoencoderKLTemporalDecoder"},
-3223 },
-3224 # file_256=[],
-3225 # layer_b3=[],
-3226 # layer_256=[],
-3227 )
-3228 )
-3229 mir_db.add(
-3230 mir_entry(
-3231 domain="info",
-3232 arch="vae",
-3233 series="kl",
-3234 comp=sdxl_series,
-3235 repo="madebyollin/sdxl-vae-fp16-fix",
-3236 pkg={
-3237 0: {"diffusers": "AutoencoderKL"},
-3238 },
-3239 file_256=[
-3240 "235745af8d86bf4a4c1b5b4f529868b37019a10f7c0b2e79ad0abca3a22bc6e1", # modelspec sai
-3241 "1b909373b28f2137098b0fd9dbc6f97f8410854f31f84ddc9fa04b077b0ace2c", # diffusers
-3242 "78f6189c8492013e3cac81637a1f657f790a237387f8a9dfd6bfa5fee28eb646", # ssd1b diffusers
-3243 "6353737672c94b96174cb590f711eac6edf2fcce5b6e91aa9d73c5adc589ee48", # ssd1b diffusers fp16
-3244 "bcb60880a46b63dea58e9bc591abe15f8350bde47b405f9c38f4be70c6161e68", # kolors fp16
-3245 "1598f3d24932bcfe6634e8b618ea1e30ab1d57f5aad13a6d2de446d2199f2341", # vega / lumina next sft d / auraflow
-3246 "703abdcd7c389316b5128faa9b750a530ea1680b453170b27afebac5e4db30c4", # pixart a
-3247 "98a14dc6fe8d71c83576f135a87c61a16561c9c080abba418d2cc976ee034f88", # hyd 1.1
-3248 ],
-3249 layer_b3=[
-3250 "bd5b356b509814025a9cf692710b87116d4fcd0e30a8232ed1db133e908d0e74", # modelspec sai
-3251 "9106380403dee83238af63ff1738396d2fdff9f6d78d0d9c1d0bf770ae4294d0", # diffusers
-3252 # "245070a60a25ca080cb4951220c3fb1503da43829930d5f6f7a6770b491eafe1",
-3253 # "50e65a628b5fe379798d8956e4a4e1d4b105c84b329f088d577f7f28c22abc49", # diffusers fp16 matches sd1
-3254 ],
-3255 layer_256=[
-3256 "c9399a4cd39a180a0bb2af96a8297b9330541e090c21e83317cebb2f7cc651da", # modelspec sai
-3257 "2240ae134a3b983abf45200c198f07e3d8068012fbbd2f658bbaa1fd6a0629c0", # diffusers
-3258 # "35641f65ad7ea600cb931dcab556f7503279f1d8d99eda170fe7976d48502a2a", # diffusers fp16 matches sd1 (incorrect)
-3259 ],
-3260 )
-3261 )
-3262 mir_db.add(
-3263 mir_entry(
-3264 domain="info",
-3265 arch="vae",
-3266 series="kl",
-3267 comp=sdxl_series + sdxl_comp,
-3268 pkg={
-3269 0: {"diffusers": "AutoencoderKL"},
-3270 },
-3271 file_256=[
-3272 "235745af8d86bf4a4c1b5b4f529868b37019a10f7c0b2e79ad0abca3a22bc6e1", # modelspec sai
-3273 "27ed3b02e09638568e99d4398c67bc654dde04e6c0db61fb2d21dba630e7058a", # diffusers
-3274 "eb6516ab7e1104d5d1a174a4d65c57835ae38061531d0a2192103aecfb790cc1", # diffusers fp16
-3275 "e6bb9ea85bbf7bf6478a7c6d18b71246f22e95d41bcdd80ed40aa212c33cfeff", # modelspec sai vae 0.9
-3276 ],
-3277 layer_b3=[
-3278 "bd5b356b509814025a9cf692710b87116d4fcd0e30a8232ed1db133e908d0e74", # modelspec sai
-3279 # "9106380403dee83238af63ff1738396d2fdff9f6d78d0d9c1d0bf770ae4294d0", # diffusers
-3280 # "245070a60a25ca080cb4951220c3fb1503da43829930d5f6f7a6770b491eafe1",
-3281 # "50e65a628b5fe379798d8956e4a4e1d4b105c84b329f088d577f7f28c22abc49", # diffusers fp16 matches sd1
-3282 ],
-3283 layer_256=[
-3284 "c9399a4cd39a180a0bb2af96a8297b9330541e090c21e83317cebb2f7cc651da", # modelspec sai
-3285 "2240ae134a3b983abf45200c198f07e3d8068012fbbd2f658bbaa1fd6a0629c0", # diffusers
-3286 # "35641f65ad7ea600cb931dcab556f7503279f1d8d99eda170fe7976d48502a2a", # diffusers fp16 matches sd1 (incorrect)
-3287 ],
-3288 )
-3289 )
-3290
-3291 repo = "shuttleai/shuttle-jaguar"
-3292 mir_db.add(
-3293 mir_entry(
-3294 domain="info",
-3295 arch="vae",
-3296 series="kl",
-3297 comp=make_mir_tag(repo)[0],
-3298 # no repo here, may conflict
-3299 pkg={
-3300 0: {"diffusers": "AutoencoderKL"},
-3301 },
-3302 file_256=[
-3303 "6fdfa2add4f04d94f36157cbb0197f97966b612e3f8eff4095315aefea74b904",
-3304 ], # q8,
-3305 layer_b3=[
-3306 "0ebf9b7010accc44e219e355dd24bf1e3128004093c0c1dfc06f88c0a39fdbdd",
-3307 "d0e7ef3c4af06fa08b4c0485a073e2df55f7b1e9e3ba8f7b261688bc562568f0", # q8
-3308 ],
-3309 layer_256=[
-3310 "9b28f36873ea283905094a64e1ccb7cfc2b0f0aa166201d0ca63807ac37caa7b", # q8
-3311 ],
-3312 )
-3313 )
-3314 mir_db.add(
-3315 mir_entry(
-3316 domain="info",
-3317 arch="vae",
-3318 series="kl",
-3319 comp=dev_series,
-3320 # no repo here, may conflict
-3321 pkg={
-3322 0: {"diffusers": "AutoencoderKL"},
-3323 },
-3324 file_256=[
-3325 "afc8e28272cd15db3919bacdb6918ce9c1ed22e96cb12c4d5ed0fba823529e38", # dev
-3326 "f5b59a26851551b67ae1fe58d32e76486e1e812def4696a4bea97f16604d40a3", # dev diffusers
-3327 "8c717328c8ad41faab2ccfd52ae17332505c6833cf176aad56e7b58f2c4d4c94", # lumina2
-3328 "8f53304a79335b55e13ec50f63e5157fee4deb2f30d5fae0654e2b2653c109dc", # sd3 turbo
-3329 ],
-3330 layer_b3=[
-3331 "b6db93ed78c4a10d69e80831c1b8fbc1447f04e9b3d494889ee2056b98d41f17", # diffusers
-3332 "a8a3ebdec4d7b38d65b7169d3604c19b587330e5e66f69ebf0ded56a24ec6903", # lumina2
-3333 # "245070a60a25ca080cb4951220c3fb1503da43829930d5f6f7a6770b491eafe1",
-3334 ],
-3335 layer_256=[
-3336 "7950e4f3897c75affaa5f9f3c51c88b4d9a27bfd9b05ad41c3f71d8c1c620b89",
-3337 "79d2bfe93a2ac037cdc59ccb5576e32d00d75d4741fba49fc7e82b9724928216", # diffusers
-3338 "8f084dc91fd5b481875bc9c86a4ef05e5f176896b7d31c6a5c2ce45c2e174004", # dev diffusers
-3339 "322e01bd511e20bc2a3c27cd611f81ed85f0046b7c023b5622c2c9a5b8b34f80", # lumina2
-3340 ],
-3341 )
-3342 )
-3343 mir_db.add(
-3344 mir_entry(
-3345 domain="info",
-3346 arch="vae",
-3347 series="eq",
-3348 comp=sdxl_series,
-3349 repo="KBlueLeaf/EQ-SDXL-VAE",
-3350 pkg={
-3351 0: {"diffusers": "AutoencoderKL"},
-3352 },
-3353 )
-3354 )
-3355 mir_db.add(
-3356 mir_entry(
-3357 domain="info",
-3358 arch="vae",
-3359 series="ms-lc-eq",
-3360 comp=sdxl_series,
-3361 repo="Anzhc/MS-LC-EQ-D-VR_VAE",
-3362 pkg={
-3363 0: {
-3364 "diffusers": "AutoencoderKL",
-3365 },
-3366 },
-3367 )
-3368 )
-3369 repo = "ucsd-reach/musicldm"
-3370 mir_db.add(
-3371 mir_entry(
-3372 domain="info",
-3373 arch="vae",
-3374 series="kl",
-3375 comp=make_mir_tag(repo)[0],
-3376 # no repo here, may conflict
-3377 file_256=[
-3378 "16e0c6c7c34e459c19500cc15cf538e6331db14969ea15917caa9b0966e44fd4",
-3379 ], # q8,
-3380 layer_b3=[
-3381 "c5c32b3fb3e73799838836ccce27d883254254daecd10f86ba8ddc55214014e0",
-3382 ],
-3383 layer_256=[
-3384 "1610c0ce39d1379091eb9ab2a4d14a8567e0f1a5dc6cca40fc0fa6f8e4e97c0f",
-3385 ],
-3386 )
-3387 )
-3388
-3389 mir_db.add(
-3390 mir_entry(
-3391 domain="info",
-3392 arch="vae",
-3393 series="kl",
-3394 comp=sd1_series,
-3395 pkg={
-3396 0: {"diffusers": "AutoencoderKL"},
-3397 },
-3398 file_256=[
-3399 "0b204ad0cae549e0a7e298d803d57e36363760dec71c63109c1da3e1147ec520", # ckpt ema original ema pruned
-3400 "95f26a5ab04779d5467d1fcecaf93160ffa523afe399b835b3e1bb77ff2d937a", # safetensors ema original ema pruned
-3401 "32db726da04f06c1b6b14c0043ce115cc87a501482945c5add89a40d838fcb46", # safetensors ema diffusers
-3402 "c6a580b13a5bc05a5e16e4dbb80608ff2ec251a162311590c1f34c013d7f3dab", # ckpt mse original ema pruned
-3403 "735e4c3a447a3255760d7f86845f09f937809baa529c17370d83e4c3758f3c75", # safetensors mse original ema pruned
-3404 "a1d993488569e928462932c8c38a0760b874d166399b14414135bd9c42df5815", # safetensors mse diffusers
-3405 "a2b5134f4dbc140d9c11f11cba3233099e00af40f262f136c691fb7d38d2194c", # safetensors diffusers
-3406 "4fbcf0ebe55a0984f5a5e00d8c4521d52359af7229bb4d81890039d2aa16dd7c", # safetensors fp16 diffusers
-3407 ],
-3408 layer_b3=[
-3409 "82e2dc440a23d78bb91df8c9fce069a8512da51f8f54ea29e3431f545808171e", # safetensors original
-3410 "2230487833925a104bee96e7ecfebaa4c3c43cc426c7a5b863f2584313dd4833", # safetensors diffusers
-3411 ],
-3412 layer_256=[
-3413 "e43f3a227b5ecb43a6272fa92ed6011d2e9abcadadd1032dfa7ea7f875f9d5bd", # safetensors original
-3414 "2494154245becf98891be884f943276aa3f54e9b3f0ea1042903fc15fba488f3", # safetensors diffusers
-3415 ],
-3416 )
-3417 )
-