| 29 | |
| 30 | @dataclass |
| 31 | class OSInsertConfig: |
| 32 | model_dir: Path |
| 33 | device: str = "cuda:0" |
| 34 | # Optional explicit checkpoint paths (override model_dir conventions) |
| 35 | objectstitch_ckpt_path: Path | None = None |
| 36 | objectstitch_config_path: Path | None = None |
| 37 | objectstitch_clip_dir: Path | None = None |
| 38 | sam_checkpoint: Path | None = None |
| 39 | flux_fill_path: Path | None = None |
| 40 | flux_redux_path: Path | None = None |
| 41 | ia_lora_path: Path | None = None |
| 42 | |
| 43 | |
| 44 | class OSInsertModel: |