(
self,
config_path="./configs/prompts/animation.yaml",
weight_dtype=torch.float16,
)
| 21 | |
| 22 | class AnimateController: |
| 23 | def __init__( |
| 24 | self, |
| 25 | config_path="./configs/prompts/animation.yaml", |
| 26 | weight_dtype=torch.float16, |
| 27 | ): |
| 28 | # Read pretrained weights path from config |
| 29 | self.config = OmegaConf.load(config_path) |
| 30 | self.pipeline = None |
| 31 | self.weight_dtype = weight_dtype |
| 32 | |
| 33 | def animate( |
| 34 | self, |
nothing calls this directly
no outgoing calls
no test coverage detected