jbilcke-hf HF Staff commited on
Commit
92833e4
·
1 Parent(s): 3f014c3

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -1,3 +1,7 @@
1
  # source code: https://github.com/lucataco/cog-sdxl-controlnet
 
 
2
  ARG threads=2
 
 
3
  FROM r8.im/lucataco/sdxl-controlnet@sha256:db2ffdbdc7f6cb4d6dab512434679ee3366ae7ab84f89750f8947d5594b79a47
 
1
  # source code: https://github.com/lucataco/cog-sdxl-controlnet
2
+ # see doc at:
3
+ # https://github.com/replicate/cog/blob/main/docs/yaml.md
4
  ARG threads=2
5
+ ARG gpus=all
6
+
7
  FROM r8.im/lucataco/sdxl-controlnet@sha256:db2ffdbdc7f6cb4d6dab512434679ee3366ae7ab84f89750f8947d5594b79a47