Remove ipex import in fastchat loader (#12984)
This commit is contained in:
		
							parent
							
								
									6f634b41da
								
							
						
					
					
						commit
						5bdf57327d
					
				
					 1 changed files with 0 additions and 1 deletions
				
			
		| 
						 | 
				
			
			@ -90,7 +90,6 @@ def load_model(
 | 
			
		|||
        model = model.eval()
 | 
			
		||||
 | 
			
		||||
    if device == "xpu":
 | 
			
		||||
        import intel_extension_for_pytorch as ipex
 | 
			
		||||
        print("Convert model to half precision...")
 | 
			
		||||
        model = model.half()
 | 
			
		||||
        model = model.to('xpu')
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in a new issue