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
|
# for the web UI
|
||||||
"gradio",
|
"gradio",
|
||||||
|
"nltk",
|
||||||
|
"sageattention==1.0.6",
|
||||||
],
|
],
|
||||||
extras_require = {
|
extras_require = {
|
||||||
"all": [
|
"all": [
|
||||||
|
@ -86,7 +88,6 @@ setup(
|
||||||
|
|
||||||
# attention helpers
|
# attention helpers
|
||||||
"xformers",
|
"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
|
# "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
|
# other audio backend that doesn't prove fruitful
|
||||||
|
|
Loading…
Reference in New Issue
Block a user