hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

get attention optimizations to work

Browse Source
This commit is contained in:
AUTOMATIC1111
2023-07-13 09:30:33 +03:00
parent b717eb7e56
commit ac4ccfa136
4 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/launch_utils.py
View File

@@ -239,6 +239,7 @@ def mute_sdxl_imports():
sys.modules['sgm.data'] = module
def prepare_environment():
torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu118")
torch_command = os.environ.get('TORCH_COMMAND', f"pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url {torch_index_url}")