Setup sshd for dstack images

dstackai · Aug 1, 2023 · f9e2b13 · f9e2b13
1 parent 80af494
commit f9e2b13
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cli/dstack/_internal/configurators/task.py b/cli/dstack/_internal/configurators/task.py
@@ -32,6 +32,8 @@ def build_commands(self) -> List[str]:
 
     def setup(self) -> List[str]:
         commands = []
+        if self.conf.image is None:
+            commands += self.sshd.get_setup_commands()
         commands += self.conf.setup
         return commands