diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 2 |
1 files changed, 2 insertions, 0 deletions
@@ -263,6 +263,8 @@ def create_scheduler(config, scheduler: str, subscheduler: Optional[str] = None) | |||
263 | config, | 263 | config, |
264 | solver_p=create_scheduler(config, subscheduler), | 264 | solver_p=create_scheduler(config, subscheduler), |
265 | ) | 265 | ) |
266 | else: | ||
267 | raise ValueError(f"Unknown scheduler \"{scheduler}\"") | ||
266 | 268 | ||
267 | 269 | ||
268 | def create_pipeline(model, dtype): | 270 | def create_pipeline(model, dtype): |