298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-06-15 15:55:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Import intel_extension_for_pytorch as ipex

Browse Source
This commit is contained in:
藍+85CD 2023-04-06 12:27:22 +08:00
parent 1ced2bdd2d
commit 84b9c0ac2f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -87,7 +87,7 @@ except:
pass pass
try: try:
import intel_extension_for_pytorch import intel_extension_for_pytorch as ipex
if torch.xpu.is_available(): if torch.xpu.is_available():
vram_state = XPU vram_state = XPU
except: except: