forgot to add NTLK as a dependency, promoted sageattn as a default dependency since it works fine enough and seems agnostic
This commit is contained in:
parent
93d27be539
commit
c66a53492c
3
setup.py
3
setup.py
|
@ -73,6 +73,8 @@ setup(
|
|||
|
||||
# for the web UI
|
||||
"gradio",
|
||||
"nltk",
|
||||
"sageattention==1.0.6",
|
||||
],
|
||||
extras_require = {
|
||||
"all": [
|
||||
|
@ -86,7 +88,6 @@ setup(
|
|||
|
||||
# attention helpers
|
||||
"xformers",
|
||||
"sageattention==1.0.6",
|
||||
# "flash-attn" --no-build-isolation # commented out right now because I want to query this for Volta freaks like me who can't use it
|
||||
|
||||
# other audio backend that doesn't prove fruitful
|
||||
|
|
Loading…
Reference in New Issue
Block a user