args.jit -> model.jit (pytorch#1086)

Summary:
Pull Request resolved: pytorch#1086

fix typo

Reviewed By: xuzhao9

Differential Revision: D38445415

fbshipit-source-id: 63ff86560e5127e381b34db7172faeab90b9f0e4

Author: davidberard98

torchbenchmark/util/extra_args.py

@@ -133,7 +133,7 @@ def apply_opt_args(model: 'torchbenchmark.util.model.BenchmarkModel', args: argp
         import torch
         model.add_context(lambda: torch.jit.fuser(args.fuser))
     if args.fx2trt:
-        if args.jit:
+        if model.jit:
             raise NotImplementedError("fx2trt with JIT is not available.")
         module, exmaple_inputs = model.get_module()
         fp16 = not (model.dargs.precision == "fp32")