Spaces:

MandlaZwane
/

Shanks

Build error

MandlaZwane commited on Jan 22, 2025

Commit

e6adc59

verified ·

1 Parent(s): d47453c

Update setup_dependencies.py

Files changed (1) hide show

setup_dependencies.py CHANGED Viewed

@@ -4,13 +4,13 @@ def install_dependencies():
     dependencies = [
         "packaging",
         "setuptools",
-        "flash-attn",
         "transformers==4.32.0",
         "safetensors",
         "tiktoken",
         "flask",
         "torch",
-        "gunicorn==20.1.0"
     ]
     for package in dependencies:

     dependencies = [
         "packaging",
         "setuptools",
         "transformers==4.32.0",
         "safetensors",
         "tiktoken",
         "flask",
         "torch",
+        "gunicorn==20.1.0",
+        "flash-attn"
     ]
     for package in dependencies: