load components from modular index

blocks.py

@@ -9,6 +9,7 @@ from diffusers.modular_pipelines import (
     ComponentSpec,
     InputParam,
     LoopSequentialPipelineBlocks,
+    ModularPipeline,
     ModularPipelineBlocks,
     OutputParam,
     SequentialPipelineBlocks,
@@ -480,14 +481,12 @@ class YuE2Blocks(SequentialPipelineBlocks):
         if components_repo is not None:
             if not trust_components_code:
                 raise ValueError("Set trust_components_code=True to load code from the model repository")
-            for name, class_name in (
-                ("transformer", "YuE2TransformerModel"),
-                ("vae", "YuE2VAE"),
-                ("tokenizer", "YuE2Tokenizer"),
-            ):
+            index = ModularPipeline.load_config(components_repo, revision=components_revision)
+            for name in ("transformer", "vae", "tokenizer"):
+                module_name, class_name, _ = index[name]
                 self.component_types[name] = get_class_from_dynamic_module(
                     components_repo,
-                    module_file=f"{name}.py",
+                    module_file=f"{module_name}.py",
                     class_name=class_name,
                     revision=components_revision,
                     trust_remote_code=True,