mirror of https://github.com/bmaltais/kohya_ss
Update torch version and validation output
parent
0d2a9765b4
commit
67f5f7b156
|
|
@ -1,9 +1,9 @@
|
|||
# Custom index URL for specific packages
|
||||
--extra-index-url https://download.pytorch.org/whl/cu124
|
||||
|
||||
torch==2.4.1+cu124
|
||||
torchvision==0.19.1+cu124
|
||||
xformers==0.0.28.post1
|
||||
torch==2.5.0+cu124
|
||||
torchvision==0.20.0+cu124
|
||||
xformers==0.0.28.post2
|
||||
|
||||
bitsandbytes==0.44.0
|
||||
tensorboard==2.15.2
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
# Custom index URL for specific packages
|
||||
--extra-index-url https://download.pytorch.org/whl/rocm6.1
|
||||
torch==2.4.0+rocm6.1
|
||||
torchvision==0.19.0+rocm6.1
|
||||
torch==2.5.0+rocm6.1
|
||||
torchvision==0.20.0+rocm6.1
|
||||
|
||||
tensorboard==2.14.1
|
||||
tensorflow-rocm==2.14.0.600
|
||||
|
|
|
|||
|
|
@ -1,8 +1,8 @@
|
|||
# Custom index URL for specific packages
|
||||
--extra-index-url https://download.pytorch.org/whl/cu124
|
||||
|
||||
torch==2.4.1+cu124
|
||||
torchvision==0.19.1+cu124
|
||||
xformers==0.0.28.post1
|
||||
torch==2.5.0+cu124
|
||||
torchvision==0.20.0+cu124
|
||||
xformers==0.0.28.post2
|
||||
|
||||
-r requirements_windows.txt
|
||||
|
|
@ -1,12 +1,12 @@
|
|||
--extra-index-url https://download.pytorch.org/whl/cu124
|
||||
torch==2.4.1+cu124
|
||||
torchvision==0.19.1+cu124
|
||||
torch==2.5.0+cu124
|
||||
torchvision==0.20.0+cu124
|
||||
xformers==0.0.28.post2
|
||||
|
||||
bitsandbytes==0.44.0
|
||||
tensorboard==2.14.1
|
||||
tensorflow==2.14.0 wheel
|
||||
tensorrt
|
||||
onnxruntime-gpu==1.17.1
|
||||
xformers==0.0.28.post1
|
||||
|
||||
-r requirements.txt
|
||||
|
|
|
|||
|
|
@ -161,12 +161,17 @@ def install_requirements_inbulk(
|
|||
log.info(f"Installing/Validating requirements from {requirements_file}...")
|
||||
|
||||
optional_parm += " -U" if upgrade else ""
|
||||
optional_parm += " --quiet" if not show_stdout else ""
|
||||
|
||||
cmd = f"pip install -r {requirements_file} {optional_parm}"
|
||||
if not show_stdout:
|
||||
cmd += " --quiet"
|
||||
cmd = [sys.executable, "-m", "pip", "install", "-r", requirements_file] + optional_parm.split()
|
||||
|
||||
if sys.platform.startswith("win32"):
|
||||
cmd += ["|", "findstr", "/V", "Requirement already satisfied"]
|
||||
else:
|
||||
cmd += ["|", "grep", "-v", "Requirement already satisfied"]
|
||||
|
||||
run_cmd(cmd)
|
||||
|
||||
log.info(f"Requirements from {requirements_file} installed/validated.")
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -178,7 +178,7 @@ def main():
|
|||
requirements_file = args.requirements or "requirements_pytorch_windows.txt"
|
||||
log.debug(f"Installing requirements from: {requirements_file}")
|
||||
setup_common.install_requirements_inbulk(
|
||||
requirements_file, show_stdout=False,
|
||||
requirements_file, show_stdout=True,
|
||||
# optional_parm="--index-url https://download.pytorch.org/whl/cu124"
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue