fix:add makefile and dockerfile

This commit is contained in:
SusanSHEN
2024-08-09 22:23:01 +02:00
parent f211f7712f
commit 0ee1d39509
6 changed files with 130 additions and 4 deletions

View File

@@ -23,13 +23,30 @@ with open("README.md", "r") as f:
# Required dependencies
REQUIRED_PACKAGES = [
"torch>=2.3.1",
"torch>=2.3.1",
"torchvision>=0.18.1",
"transformers",
"numpy>=1.24.4",
"tqdm>=4.66.1",
"hydra-core>=1.3.2",
"iopath>=0.1.10",
"pillow>=9.4.0",
"pillow>=9.4.0",
"huggingface_hub",
"diffusers[torch]==0.15.1",
"onnxruntime==1.14.1",
"onnx==1.13.1",
"ipykernel==6.16.2",
"scipy",
"gradio",
"openai",
"matplotlib>=3.9.1",
"opencv-python>=4.7.0",
"dds_cloudapi_sdk",
"addict",
"yapf",
"timm",
"supervision>=0.22.0",
"pycocotools",
]
EXTRA_PACKAGES = {