(request: dict[str, Any], noise_file: str)
| 219 | |
| 220 | |
| 221 | def make_generation_config(request: dict[str, Any], noise_file: str) -> GenerationConfig: |
| 222 | batch_size = int(request.get("batch_size", 1)) |
| 223 | return GenerationConfig( |
| 224 | batch_size=batch_size, |
| 225 | allow_lm_batch=False, |
| 226 | use_random_seed=False, |
| 227 | seeds=[int(request.get("seed", 0))], |
| 228 | noise_file=noise_file or None, |
| 229 | lm_batch_chunk_size=1, |
| 230 | constrained_decoding_debug=False, |
| 231 | audio_format="wav", |
| 232 | ) |
| 233 | |
| 234 | |
| 235 | def request_needs_llm(request: dict[str, Any]) -> bool: |
no test coverage detected