Hugging Face's logo

Spaces:
banao-tech
/
omniparser-fast
Runtime error

App Files Community
banao-tech commited on
Commit
00a6698
Β·
verified Β·
1 Parent(s): aa383a1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  # Base image
2
- FROM huggingface/transformers-pytorch-gpu:4.33.2
3
 
4
  # Install system dependencies
5
  RUN apt-get update && apt-get install -y \
 
1
  # Base image
2
+ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
3
 
4
  # Install system dependencies
5
  RUN apt-get update && apt-get install -y \