mirror of
https://github.com/NVIDIA/Model-Optimizer.git
synced 2026-11-10 09:34:01 +08:00
Update for 0.31.0 release
This commit is contained in:
@@ -134,5 +134,5 @@ def infer(pipe):
|
||||
|
||||
def prepare(pipe, config_list):
|
||||
model = get_model(pipe)
|
||||
assert model.__class__ in CACHED_PIPE.keys(), f"{model.__class__} is not supported!"
|
||||
assert model.__class__ in CACHED_PIPE, f"{model.__class__} is not supported!"
|
||||
cachify(model, config_list, CACHED_PIPE[model.__class__])
|
||||
|
||||
Reference in New Issue
Block a user