Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 8 additions & 12 deletions comfy_cli/command/install.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from rich.panel import Panel
from rich.prompt import Confirm

from comfy_cli import constants, ui, utils
from comfy_cli import constants, ui
from comfy_cli.command.custom_nodes.command import update_node_id_cache
from comfy_cli.command.github.pr_info import PRInfo
from comfy_cli.constants import GPU_OPTION
Expand Down Expand Up @@ -96,28 +96,24 @@ def pip_install_comfyui_dependencies(
base_command,
check=False,
)
# Beta support for intel arch based on this PR: https://github.com/comfyanonymous/ComfyUI/pull/3439
# Update installation to use upstream torch xpu. ipex is no longer needed for Intel Arc GPUs
# https://github.com/comfyanonymous/ComfyUI/pull/7767
if gpu == GPU_OPTION.INTEL_ARC:
pip_url = [
"--extra-index-url",
"https://pytorch-extension.intel.com/release-whl/stable/xpu/us/",
"https://download.pytorch.org/whl/xpu",
]
utils.install_conda_package("libuv")

# TODO: wrap pip install in a function
subprocess.run(
[sys.executable, "-m", "pip", "install", "mkl", "mkl-dpcpp"],
check=True,
)
result = subprocess.run(
[
sys.executable,
"-m",
"pip",
"install",
"torch==2.1.0.post2",
"torchvision==0.16.0.post2",
"torchaudio==2.1.0.post2",
"intel-extension-for-pytorch==2.1.30",
"torch",
"torchvision",
"torchaudio",
]
+ pip_url,
check=False,
Expand Down
Loading