Fix missing pydantic updates in legacy mii code (#524)

Signed-off-by: Logan Adams <[email protected]>
deepspeedai · Feb 7, 2025 · d2c93b3 · d2c93b3
1 parent 7185697
commit d2c93b3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mii/legacy/deployment.py b/mii/legacy/deployment.py
@@ -37,12 +37,12 @@ def support_legacy_api(
     }
     # TODO do this in a single for loop
     for key, val in mii_config.items():
-        if key not in MIIConfig.fields.keys():
+        if key not in MIIConfig.model_fields.keys():
             model_config[key] = val
     mii_config = {
         k: v
         for k,
-        v in mii_config.items() if k in MIIConfig.__dict__["__fields__"]
+        v in mii_config.items() if k in MIIConfig.model_fields.keys()
     }
     mii_config["version"] = version
     mii_config["deployment_type"] = deployment_type