Skip to content

Commit

Permalink
limit numpy<2 override to windows only
Browse files Browse the repository at this point in the history
  • Loading branch information
telamonian committed Aug 30, 2024
1 parent 846a5c5 commit 56c2512
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions comfy_cli/uv.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,9 @@
from typing import Any, Optional, Union, cast

from comfy_cli import ui
from comfy_cli.constants import GPU_OPTION
from comfy_cli.constants import GPU_OPTION, OS
from comfy_cli.typing import PathLike
from comfy_cli.utils import get_os


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

# TODO: remove numpy<2 override once torch is compatible with numpy>=2
f.write("numpy<2\n")
f.write("\n\n")
if get_os == OS.WINDOWS:
f.write("numpy<2\n")
f.write("\n\n")

completed = DependencyCompiler.Compile(
cwd=self.cwd,
Expand Down

0 comments on commit 56c2512

Please sign in to comment.