Spaces:
Runtime error
Runtime error
Commit
•
43b6675
1
Parent(s):
8d3da67
fix xformers
Browse files- Dockerfile +2 -0
- requirements.txt +0 -1
Dockerfile
CHANGED
|
@@ -25,6 +25,8 @@ RUN pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url h
|
|
| 25 |
|
| 26 |
RUN pip install -r requirements.txt
|
| 27 |
|
|
|
|
|
|
|
| 28 |
RUN conda install -c conda-forge cudatoolkit-dev -y
|
| 29 |
ENV CUDA_HOME=$CONDA_PREFIX/pkgs/cuda-toolkit/
|
| 30 |
RUN pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"
|
|
|
|
| 25 |
|
| 26 |
RUN pip install -r requirements.txt
|
| 27 |
|
| 28 |
+
RUN pip install xformers==0.0.22.post7 --index-url https://download.pytorch.org/whl/cu118
|
| 29 |
+
|
| 30 |
RUN conda install -c conda-forge cudatoolkit-dev -y
|
| 31 |
ENV CUDA_HOME=$CONDA_PREFIX/pkgs/cuda-toolkit/
|
| 32 |
RUN pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"
|
requirements.txt
CHANGED
|
@@ -33,7 +33,6 @@ triton==2.1.0
|
|
| 33 |
torchdata==0.7.0
|
| 34 |
wandb
|
| 35 |
invisible-watermark
|
| 36 |
-
xformers
|
| 37 |
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
|
| 38 |
-e git+https://github.com/openai/CLIP.git@main#egg=clip
|
| 39 |
-e git+https://github.com/Stability-AI/datapipelines.git@main#egg=sdata
|
|
|
|
| 33 |
torchdata==0.7.0
|
| 34 |
wandb
|
| 35 |
invisible-watermark
|
|
|
|
| 36 |
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
|
| 37 |
-e git+https://github.com/openai/CLIP.git@main#egg=clip
|
| 38 |
-e git+https://github.com/Stability-AI/datapipelines.git@main#egg=sdata
|