(self)
| 117 | def load_pipeline(self) -> None: ... |
| 118 | |
| 119 | def load_captioner(self) -> None: |
| 120 | if self.args.captioner == "none": |
| 121 | self.captioner = EmptyCaptioner(self.args.device) |
| 122 | elif self.args.captioner == "llava": |
| 123 | assert LLAVA_AVAILABLE, "llava is not available in your environment." |
| 124 | self.captioner = LLaVACaptioner(self.args.device, self.args.llava_bit) |
| 125 | elif self.args.captioner == "ram": |
| 126 | assert RAM_AVAILABLE, "ram is not available in your environment." |
| 127 | self.captioner = RAMCaptioner(self.args.device) |
| 128 | else: |
| 129 | raise ValueError(f"unsupported captioner: {self.args.captioner}") |
| 130 | |
| 131 | def setup(self) -> None: |
| 132 | self.save_dir = self.args.output |
no test coverage detected