RFD4-Proteina-Diffusers / model_index.json

Split the scheduler per modality, move recycle_depths to the pipeline config, rename feature-embedding keys

5db5138 verified 2 days ago

817 Bytes

{
  "_class_name": "RFD4ProteinaPipeline",
  "_diffusers_version": "0.42.0.dev0",
  "bb_ca_scheduler": [
    "diffusers",
    "RFD4ProteinaScheduler"
  ],
  "local_latents_scheduler": [
    "diffusers",
    "RFD4ProteinaScheduler"
  ],
  "recycle_depths": [
    6,
    6,
    6,
    6,
    6,
    5,
    5,
    5,
    5,
    5,
    5,
    5,
    5,
    5,
    5,
    4,
    4,
    4,
    4,
    4,
    3,
    3,
    3,
    3,
    3,
    3,
    3,
    3,
    3,
    3,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    1,
    1,
    1,
    1,
    1,
    1,
    1,
    1,
    1,
    1
  ],
  "transformer": [
    "diffusers",
    "RFD4ProteinaTransformer"
  ],
  "vae": [
    "diffusers",
    "RFD4ProteinaAutoencoder"
  ]
}

	{
	"_class_name": "RFD4ProteinaPipeline",
	"_diffusers_version": "0.42.0.dev0",
	"bb_ca_scheduler": [
	"diffusers",
	"RFD4ProteinaScheduler"
	],
	"local_latents_scheduler": [
	"diffusers",
	"RFD4ProteinaScheduler"
	],
	"recycle_depths": [
	6,
	6,
	6,
	6,
	6,
	5,
	5,
	5,
	5,
	5,
	5,
	5,
	5,
	5,
	5,
	4,
	4,
	4,
	4,
	4,
	3,
	3,
	3,
	3,
	3,
	3,
	3,
	3,
	3,
	3,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	2,
	1,
	1,
	1,
	1,
	1,
	1,
	1,
	1,
	1,
	1
	],
	"transformer": [
	"diffusers",
	"RFD4ProteinaTransformer"
	],
	"vae": [
	"diffusers",
	"RFD4ProteinaAutoencoder"
	]
	}