Dynamically load the generator class from generator.py.
(manifest: dict)
| 71 | # ------------------------------------------------------------------ # |
| 72 | |
| 73 | def load_generator(manifest: dict): |
| 74 | """Dynamically load the generator class from generator.py.""" |
| 75 | spec = importlib.util.spec_from_file_location( |
| 76 | "generator", EXT_DIR / "generator.py" |
| 77 | ) |
| 78 | mod = importlib.util.module_from_spec(spec) |
| 79 | sys.modules["generator"] = mod |
| 80 | spec.loader.exec_module(mod) |
| 81 | return getattr(mod, manifest["generator_class"]) |
| 82 | |
| 83 | |
| 84 | def _select_node(manifest: dict, model_dir_override: str) -> dict: |