updated env setup

GaParmar · Aug 10, 2024 · 2595e1b · 2595e1b
1 parent 5043112
commit 2595e1b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/environment.yaml b/environment.yaml
@@ -23,7 +23,6 @@ dependencies:
 
       - tqdm>=4.65.0
       - transformers==4.35.2
-      - triton==2.0.0
       - urllib3<1.27,>=1.25.4
       - xformers>=0.0.20
       - streamlit-keyup==0.2.0

diff --git a/requirements.txt b/requirements.txt
@@ -16,7 +16,6 @@ torchvision>=0.15.2
 
 tqdm>=4.65.0
 transformers==4.35.2
-triton==2.0.0
 urllib3<1.27,>=1.25.4
 xformers>=0.0.20
 streamlit-keyup==0.2.0