{
    "name": "EveryDream2 Dev Container",
	// This Dockerfile requires buildx to be enabled by default
	// `docker buildx install`
	"build": {
		"dockerfile": "../../docker/Dockerfile",
		"target": "runtime"
	},
	"postStartCommand": "/start.sh",
	"forwardPorts": [8888, 6006],
	"containerEnv": {
		"JUPYTER_PASSWORD": "EveryDream"
	},
	
	// I don't have a GPU to test this with
	"runArgs": [
		"--gpus",
		"all"
	],

	// Mimic RunPod/Vast setup
	"workspaceMount": "source=${localWorkspaceFolder},target=/workspace/EveryDream2trainer,type=bind",
	"workspaceFolder": "/workspace/EveryDream2trainer",
	"customizations": {
		"vscode": {
			"extensions": [
				"ms-python.python"
			]
		}
	}
}