Hugging Face's logo

Spaces:
duongve
/
Spatial_Control_for_SD
Running

App Files Community
duongve commited on
Commit
a2f8603
·
verified ·
1 Parent(s): 7ef93e7

Upload Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  # Dockerfile Public T4
2
 
3
- FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
4
  ENV DEBIAN_FRONTEND noninteractive
5
 
6
  WORKDIR /content
 
1
  # Dockerfile Public T4
2
 
3
+ FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn8-devel
4
  ENV DEBIAN_FRONTEND noninteractive
5
 
6
  WORKDIR /content