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):
12081208 _validate_args (args )
12091209 e_mgr = _prepare_for_llama_export (args )
12101210 model = (
1211- e_mgr .model .eval ().to (device = "cuda" ) # pyre-ignore
1211+ e_mgr .model .eval ().to (device = "cuda" )
12121212 if torch .cuda .is_available ()
12131213 else e_mgr .model .eval ().to (device = "cpu" )
12141214 )
Original file line number Diff line number Diff line change @@ -127,7 +127,7 @@ def detect_ambiguity(gm):
127127 Check every node's output tensor dim_order and raise if it is ambiguous for a list of formats.
128128 """
129129
130- def get_tensors (node : torch .fx .Node ) -> torch .Tensor :
130+ def get_tensors (node : torch .fx .Node ) -> List [ torch .Tensor ] :
131131 val = node .meta ["val" ]
132132 if isinstance (val , torch .Tensor ):
133133 return [val ]
You can’t perform that action at this time.
0 commit comments