File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -364,7 +364,7 @@ def load_config(
364364            if  subfolder  is  not None :
365365                raise  ValueError ("DDUF file only allow for 1 level of directory. Please check the DDUF structure" )
366366            # paths inside a DDUF file must always be "/" 
367-             config_file  =  "/" .join ([pretrained_model_name_or_path , cls .config_name ])
367+             config_file  =  cls . config_name   if   pretrained_model_name_or_path   ==   ""   else   "/" .join ([pretrained_model_name_or_path , cls .config_name ])
368368            if  config_file  not  in dduf_entries :
369369                raise  ValueError (
370370                    f"We did not manage to find the file { config_file } { dduf_entries .keys ()}  
Original file line number Diff line number Diff line change @@ -299,8 +299,9 @@ def _get_model_file(
299299    if  dduf_entries :
300300        if  subfolder  is  not None :
301301            raise  ValueError ("DDUF file only allow for 1 level of directory. Please check the DDUF structure" )
302-         if  "/" .join ([pretrained_model_name_or_path , weights_name ]) in  dduf_entries :
303-             return  "/" .join ([pretrained_model_name_or_path , weights_name ])
302+         model_file  =  weights_name  if  pretrained_model_name_or_path  ==  ""  else  "/" .join ([pretrained_model_name_or_path , weights_name ])
303+         if  model_file  in  dduf_entries :
304+             return  model_file 
304305        else :
305306            raise  EnvironmentError (f"Error no file named { weights_name } { dduf_entries .keys ()}  )
306307    elif  os .path .isfile (pretrained_model_name_or_path ):
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments