Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
S
Stable Diffusion Webui
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Stable Diffusion Webui
Commits
96cba45d
Commit
96cba45d
authored
May 18, 2023
by
Sakura-Luna
Committed by
GitHub
May 18, 2023
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Modify xformers instead of pytorch
parent
9a9557ec
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
launch.py
launch.py
+3
-2
No files found.
launch.py
View file @
96cba45d
...
@@ -237,7 +237,8 @@ def run_extensions_installers(settings_file):
...
@@ -237,7 +237,8 @@ def run_extensions_installers(settings_file):
def
prepare_environment
():
def
prepare_environment
():
torch_command
=
os
.
environ
.
get
(
'TORCH_COMMAND'
,
"pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118"
)
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}"
)
requirements_file
=
os
.
environ
.
get
(
'REQS_FILE'
,
"requirements_versions.txt"
)
requirements_file
=
os
.
environ
.
get
(
'REQS_FILE'
,
"requirements_versions.txt"
)
xformers_package
=
os
.
environ
.
get
(
'XFORMERS_PACKAGE'
,
'xformers==0.0.17'
)
xformers_package
=
os
.
environ
.
get
(
'XFORMERS_PACKAGE'
,
'xformers==0.0.17'
)
...
@@ -292,7 +293,7 @@ def prepare_environment():
...
@@ -292,7 +293,7 @@ def prepare_environment():
if
not
is_installed
(
"xformers"
):
if
not
is_installed
(
"xformers"
):
exit
(
0
)
exit
(
0
)
elif
platform
.
system
()
==
"Linux"
:
elif
platform
.
system
()
==
"Linux"
:
run_pip
(
f
"install {xformers_package}"
,
"xformers"
)
run_pip
(
f
"install
-U -I --no-deps
{xformers_package}"
,
"xformers"
)
if
not
is_installed
(
"pyngrok"
)
and
args
.
ngrok
:
if
not
is_installed
(
"pyngrok"
)
and
args
.
ngrok
:
run_pip
(
"install pyngrok"
,
"ngrok"
)
run_pip
(
"install pyngrok"
,
"ngrok"
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment