BarBar288 commited on
Commit
0ff3338
·
verified ·
1 Parent(s): 5e3f3e0

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -10,7 +10,7 @@ COPY . /home/user/app
10
  # Install any needed packages specified in requirements.txt
11
  RUN pip install --no-cache-dir -r requirements.txt
12
  # Allocate Resources
13
- docker run -p 7860:7860 --env ACC_TOKEN=your_huggingface_token --cpus=8 --memory=16g your-image-name
14
  # Set the environment variable for Hugging Face access token
15
  ARG ACC_TOKEN
16
  ENV AccToken=${ACC_TOKEN}
 
10
  # Install any needed packages specified in requirements.txt
11
  RUN pip install --no-cache-dir -r requirements.txt
12
  # Allocate Resources
13
+ docker run -p 7860:7860 --env ACC_TOKEN=your_huggingface_token --cpus=32 --memory=200g --gpus=10 your-image-name
14
  # Set the environment variable for Hugging Face access token
15
  ARG ACC_TOKEN
16
  ENV AccToken=${ACC_TOKEN}