Skip to content

Commit a2d1edb

Browse files
committed
Update pytorch launcher readme
1 parent 47da7de commit a2d1edb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tools/accuracy_checker/accuracy_checker/launcher/pytorch_launcher_readme.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ For enabling PyTorch launcher you need to add `framework: pytorch` in launchers
1717
* `batch` - batch size for running model (Optional, default 1).
1818
* `use_torch_compile` - boolean, use torch.compile to optimize the module code (Optional, default `False`)
1919
* `torch_compile_kwargs` - dictionary of keyword arguments to pass to torch.compile (Optional, default `{}`)
20+
* `transformers_class` - transformers class name to load pre-trained model with `module` name. (Optional).
2021

2122
In turn if you model has several inputs you need to specify them in config, using specific parameter: `inputs`.
2223
Each input description should has following info:

0 commit comments

Comments
 (0)