Hugging Face's logo

Spaces:
blanchon
/
TiM
Sleeping

App Files Community
blanchon commited on
Commit
ebc839f
·
1 Parent(s): fdd45ab

Update

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +1 -1
Dockerfile CHANGED
@@ -58,7 +58,7 @@ COPY --chown=user requirements.txt .
58
 
59
  # Install Python packages with uv caching
60
  RUN --mount=type=cache,target=$UV_CACHE_DIR,uid=1000,gid=1000 \
61
- /uv pip install torch==2.8.0
62
 
63
  # Install build dependencies for flash-attn
64
  RUN --mount=type=cache,target=$UV_CACHE_DIR,uid=1000,gid=1000 \
 
58
 
59
  # Install Python packages with uv caching
60
  RUN --mount=type=cache,target=$UV_CACHE_DIR,uid=1000,gid=1000 \
61
+ /uv pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cu118
62
 
63
  # Install build dependencies for flash-attn
64
  RUN --mount=type=cache,target=$UV_CACHE_DIR,uid=1000,gid=1000 \
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  gradio>=4.0.0
2
  spaces>=0.28.0
3
- torch==2.8.0
4
  torchvision
5
  diffusers
6
  transformers>=4.25.0
 
1
  gradio>=4.0.0
2
  spaces>=0.28.0
3
+ torch
4
  torchvision
5
  diffusers
6
  transformers>=4.25.0