cog.yaml 485 B

12345678910111213141516171819202122
  1. # This file is used for constructing replicate env
  2. image: "r8.im/tencentarc/realesrgan"
  3. build:
  4. gpu: true
  5. python_version: "3.8"
  6. system_packages:
  7. - "libgl1-mesa-glx"
  8. - "libglib2.0-0"
  9. python_packages:
  10. - "torch==1.7.1"
  11. - "torchvision==0.8.2"
  12. - "numpy==1.21.1"
  13. - "lmdb==1.2.1"
  14. - "opencv-python==4.5.3.56"
  15. - "PyYAML==5.4.1"
  16. - "tqdm==4.62.2"
  17. - "yapf==0.31.0"
  18. - "basicsr==1.4.2"
  19. - "facexlib==0.2.5"
  20. predict: "cog_predict.py:Predictor"