Remove ipex import in fastchat loader (#12984)

This commit is contained in:
Yuwen Hu 2025-03-20 18:29:00 +08:00 committed by GitHub
parent 6f634b41da
commit 5bdf57327d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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')