Skip to content

Commit de8e8e3

Browse files
authored
Fix xpu Pytorch nightly build from calling optimize which doesn't exist. (comfyanonymous#4978)
1 parent a1e71cf commit de8e8e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/model_management.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,7 @@ def model_load(self, lowvram_model_memory=0, force_patch_weights=False):
326326
self.model_unload()
327327
raise e
328328

329-
if is_intel_xpu() and not args.disable_ipex_optimize and self.real_model is not None:
329+
if is_intel_xpu() and not args.disable_ipex_optimize and 'ipex' in globals() and self.real_model is not None:
330330
with torch.no_grad():
331331
self.real_model = ipex.optimize(self.real_model.eval(), inplace=True, graph_mode=True, concat_linear=True)
332332

0 commit comments

Comments
 (0)