Skip to content

Commit

Permalink
Update requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
bmaltais committed Jun 23, 2023
1 parent 10191fc commit e3ce447
Show file tree
Hide file tree
Showing 6 changed files with 5 additions and 2 deletions.
1 change: 1 addition & 0 deletions requirements_linux.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
xformers==0.0.20
tensorboard==2.12.1 tensorflow==2.12.0
-r requirements.txt

# accelerate==0.19.0
# albumentations==1.3.0
Expand Down
1 change: 1 addition & 0 deletions requirements_macos_amd64.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
torch==2.0.0 torchvision==0.15.1 -f https://download.pytorch.org/whl/cpu/torch_stable.html
xformers
tensorflow-macos tensorboard==2.12.1
-r requirements.txt

# accelerate==0.15.0
# albumentations==1.3.0
Expand Down
1 change: 1 addition & 0 deletions requirements_macos_arm64.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
torch==2.0.0 torchvision==0.15.1 -f https://download.pytorch.org/whl/cpu/torch_stable.html
xformers
tensorflow-metal tensorboard==2.12.1
-r requirements.txt

# accelerate==0.15.0
# albumentations==1.3.0
Expand Down
1 change: 1 addition & 0 deletions requirements_windows_torch1.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
tensorboard==2.10.1
tensorflow==2.10.1
accelerate==0.15.0
-r requirements.txt

# albumentations==1.3.0
# altair==4.2.2
Expand Down
1 change: 1 addition & 0 deletions requirements_windows_torch2.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
tensorboard==2.12.3
tensorflow==2.12.0
accelerate==0.19.0
-r requirements.txt

# albumentations==1.3.0
# altair==4.2.2
Expand Down
2 changes: 0 additions & 2 deletions setup/setup_linux.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,6 @@ def install_kohya_ss(platform_requirements_file):
setup_common.install('--upgrade pip')

setup_common.install_requirements(platform_requirements_file)

setup_common.install_requirements('requirements.txt')
# setup_common.configure_accelerate()
# run_cmd(f'accelerate config')

Expand Down

0 comments on commit e3ce447

Please sign in to comment.