File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -1208,7 +1208,7 @@ def get_llama_model(args):
1208
1208
_validate_args (args )
1209
1209
e_mgr = _prepare_for_llama_export (args )
1210
1210
model = (
1211
- e_mgr .model .eval ().to (device = "cuda" ) # pyre-ignore
1211
+ e_mgr .model .eval ().to (device = "cuda" )
1212
1212
if torch .cuda .is_available ()
1213
1213
else e_mgr .model .eval ().to (device = "cpu" )
1214
1214
)
Original file line number Diff line number Diff line change @@ -127,7 +127,7 @@ def detect_ambiguity(gm):
127
127
Check every node's output tensor dim_order and raise if it is ambiguous for a list of formats.
128
128
"""
129
129
130
- def get_tensors (node : torch .fx .Node ) -> torch .Tensor :
130
+ def get_tensors (node : torch .fx .Node ) -> List [ torch .Tensor ] :
131
131
val = node .meta ["val" ]
132
132
if isinstance (val , torch .Tensor ):
133
133
return [val ]
You can’t perform that action at this time.
0 commit comments