parent
							
								
									3e2af5ec0a
								
							
						
					
					
						commit
						f8730e8dc1
					
				
					 1 changed files with 4 additions and 0 deletions
				
			
		| 
						 | 
				
			
			@ -561,6 +561,10 @@ class _BaseAutoModelClass:
 | 
			
		|||
        else:
 | 
			
		||||
            model = model_class(config, *model_args, **kwargs)
 | 
			
		||||
 | 
			
		||||
        # rwkv model linear layers has been rescaled
 | 
			
		||||
        if model.config.model_type == "rwkv":
 | 
			
		||||
            model.layers_are_rescaled = True
 | 
			
		||||
 | 
			
		||||
        # Loading args may differ based on their usage
 | 
			
		||||
        quant_device = "meta" if bigdl_lcmu_enabled else "cpu"
 | 
			
		||||
        model = ggml_convert_low_bit(model, qtype, optimize_model, device=quant_device,
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in a new issue