summaryrefslogtreecommitdiffstats
path: root/infer.py
diff options
context:
space:
mode:
Diffstat (limited to 'infer.py')
-rw-r--r--infer.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/infer.py b/infer.py
index 8910e68..07dcd22 100644
--- a/infer.py
+++ b/infer.py
@@ -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
268def create_pipeline(model, dtype): 270def create_pipeline(model, dtype):