[CoreEngine] set the first device as the master and slave id when dep…

…loying models.
jrpizani · Nov 24, 2023 · d15b98a · d15b98a
1 parent ce3c078
commit d15b98a
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 5 deletions.
diff --git a/devops/dockerfile/device-image/Dockerfile-Release b/devops/dockerfile/device-image/Dockerfile-Release
@@ -1,6 +1,6 @@
 ARG VERSION=release
 ARG IS_BUILDING_GPU_IMAGE=0
-ARG BASE_IMAGE=public.ecr.aws/x6k8q1x9/fedml-device-image:base
+ARG BASE_IMAGE=docker.io/fedml/fedml-device-image:base
 FROM ${BASE_IMAGE}
 
 ADD ./devops/scripts/runner.sh ./fedml/runner.sh

diff --git a/python/fedml/__init__.py b/python/fedml/__init__.py
@@ -34,7 +34,7 @@
 _global_training_type = None
 _global_comm_backend = None
 
-__version__ = "0.8.12.dev36"
+__version__ = "0.8.12.dev37"
 
 
 # This is the deployment environment used for different roles (RD/PM/BD/Public Developers). Potential VALUE: local, dev, test, release

diff --git a/python/fedml/computing/scheduler/master/server_runner.py b/python/fedml/computing/scheduler/master/server_runner.py
@@ -2070,8 +2070,6 @@ def check_model_device_ready_and_deploy(self, run_id, master_device_id, slave_de
         # Generate serving devices for deploying
         serving_devices = list()
         serving_devices.append(device_master_ids[0])
-        if len(device_slave_ids) > 1:
-            device_slave_ids.pop(0)
         serving_devices.extend(device_slave_ids)
 
         # Start to deploy the model

diff --git a/python/setup.py b/python/setup.py
@@ -94,7 +94,7 @@ def finalize_options(self):
 
 setup(
     name="fedml",
-    version="0.8.12.dev36",
+    version="0.8.12.dev37",
     author="FedML Team",
     author_email="ch@fedml.ai",
     description="A research and production integrated edge-cloud library for "