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