Fetch model details and set up internal state.
(self)
| 28 | self._ready = False |
| 29 | |
| 30 | def _initialize(self): |
| 31 | """Fetch model details and set up internal state.""" |
| 32 | result = self._bytez.list.models({"modelId": self.id}) |
| 33 | media_generators = { |
| 34 | "text-to-audio", |
| 35 | "text-to-image", |
| 36 | "text-to-video", |
| 37 | "text-to-speech", |
| 38 | } |
| 39 | |
| 40 | self.details = result.output[0] if result.output else {} |
| 41 | self._is_generating_media = self.details.get("task") in media_generators |
| 42 | self._ready = True |
| 43 | |
| 44 | def run( |
| 45 | self, |