ngc command fix

huggingface · Dec 19, 2023 · b21138d · b21138d
1 parent 89918fa
commit b21138d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/autotrain/backend.py b/src/autotrain/backend.py
@@ -520,6 +520,7 @@ def create(self):
             "envs": [{"name": key, "value": value} for key, value in self.env_vars.items()],
             "jobOrder": 50,
             "jobPriority": "NORMAL",
+            "portMappings": [{"containerPort": 7860, "protocol": "HTTPS"}],
             "resultContainerMountPoint": "/results",
             "runPolicy": {"preemptClass": "RUNONCE", "totalRuntimeSeconds": 259200},
         }