Redundant `if-else`

[Andy Lo]

qlora/qlora.py

Lines 248 to 259 in 7f4e95a

 def find_all_linear_names(args, model): 
     cls = bnb.nn.Linear4bit if args.bits == 4 else (bnb.nn.Linear8bitLt if args.bits == 8 else torch.nn.Linear) 
     lora_module_names = set() 
     for name, module in model.named_modules(): 
         if isinstance(module, cls): 
             names = name.split('.') 
             lora_module_names.add(names[0] if len(names) == 1 else names[-1]) 
  
  
     if 'lm_head' in lora_module_names: # needed for 16-bit 
         lora_module_names.remove('lm_head') 
     return list(lora_module_names) 

I think the names[0] if len(names) == 1 else names[-1] expression in L254 is just redundant. Should use just names[-1].