check whether prefill stage exist
(self)
| 288 | logger.exception("Exception in async output loop: %s", e) |
| 289 | |
| 290 | def exist_prefill(self): |
| 291 | """ |
| 292 | check whether prefill stage exist |
| 293 | """ |
| 294 | if envs.ENABLE_V1_KVCACHE_SCHEDULER: |
| 295 | return self.exist_prefill_flag |
| 296 | return (self.share_inputs["seq_lens_encoder"] > 0).any().cpu().numpy().item() |
| 297 | |
| 298 | @property |
| 299 | def is_sleeping(self): |
no test coverage detected