Skip to content

Commit

Permalink
More torch 1 code cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
bmaltais committed Oct 4, 2023
1 parent 93a181f commit a756f0a
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 6 deletions.
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -627,7 +627,13 @@ ControlNet-LLLite, a novel method for ControlNet with SDXL, is added. See [docum

* 2023/10/08 (v22.1.0)
- Remove support for torch 1 to align with kohya_ss sd-scripts code base.

- Add Intel ARC GPU support with IPEX support on Linuix / WSL
- Users needs to set these manually:
* Mixed precision to BF16,
* Attention to SDPA,
* Optimizer to: AdamW (or any other non 8 bit one).
- Run setup with: `./setup.sh --use-ipex`
- Run the GUI with: `./gui.sh --use-ipex`
* 2023/10/01 (v22.0.0)
- Merging main branch of sd-scripts:
- [SAI Model Spec](https://github.com/Stability-AI/ModelSpec) metadata is now supported partially. `hash_sha256` is not supported yet.
Expand Down
10 changes: 5 additions & 5 deletions setup/setup_windows.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,11 +132,11 @@ def install_kohya_ss_torch2():
# Upgrade pip if needed
setup_common.install('--upgrade pip')

if setup_common.check_torch() == 1:
input(
f'{YELLOW}\nTorch 1 is already installed in the venv. To install Torch 2 delete the venv and re-run setup.bat\n\nHit any key to acknowledge.{RESET_COLOR}'
)
return
# if setup_common.check_torch() == 1:
# input(
# f'{YELLOW}\nTorch 1 is already installed in the venv. To install Torch 2 delete the venv and re-run setup.bat\n\nHit any key to acknowledge.{RESET_COLOR}'
# )
# return

# setup_common.install(
# 'torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118',
Expand Down

0 comments on commit a756f0a

Please sign in to comment.