Skip to content

Commit 56c2512

Browse files
committed
limit numpy<2 override to windows only
1 parent 846a5c5 commit 56c2512

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

comfy_cli/uv.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,9 @@
77
from typing import Any, Optional, Union, cast
88

99
from comfy_cli import ui
10-
from comfy_cli.constants import GPU_OPTION
10+
from comfy_cli.constants import GPU_OPTION, OS
1111
from comfy_cli.typing import PathLike
12+
from comfy_cli.utils import get_os
1213

1314

1415
def _run(cmd: list[str], cwd: PathLike, check: bool = True) -> subprocess.CompletedProcess[Any]:
@@ -370,8 +371,9 @@ def make_override(self):
370371
f.write("\n\n")
371372

372373
# TODO: remove numpy<2 override once torch is compatible with numpy>=2
373-
f.write("numpy<2\n")
374-
f.write("\n\n")
374+
if get_os == OS.WINDOWS:
375+
f.write("numpy<2\n")
376+
f.write("\n\n")
375377

376378
completed = DependencyCompiler.Compile(
377379
cwd=self.cwd,

0 commit comments

Comments
 (0)