Spaces:

edbeeching
/

protein-folding-api

Sleeping

edbeeching HF Staff commited on 14 days ago

Commit

d29824d

verified ·

1 Parent(s): 8ab4ff2

Use torch 2.6 for ESMFold loading

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
@@ -23,4 +23,3 @@ COPY . /app
 EXPOSE 7860
 CMD ["uvicorn", "folding_api_service.app:app", "--host", "0.0.0.0", "--port", "7860"]

+FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
 EXPOSE 7860
 CMD ["uvicorn", "folding_api_service.app:app", "--host", "0.0.0.0", "--port", "7860"]