Do not require sshd in task configuration for custom docker images (#609

) * Do not require sshd in task configuration for custom docker images * Setup sshd for dstack images
dstackai · Aug 1, 2023 · 5a02825 · 5a02825
1 parent 4a4de97
commit 5a02825
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cli/dstack/_internal/configurators/task.py b/cli/dstack/_internal/configurators/task.py
@@ -32,9 +32,8 @@ def build_commands(self) -> List[str]:
 
     def setup(self) -> List[str]:
         commands = []
-        if self.conf.image:
-            commands += self.sshd.get_required_commands()
-        commands += self.sshd.get_setup_commands()
+        if self.conf.image is None:
+            commands += self.sshd.get_setup_commands()
         commands += self.conf.setup
         return commands