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