fix: support more instance type for custom (#97)

yanbasic · web-flow · commit abaf04bb9c58 · 2025-04-09T17:09:36.000+08:00
* fix: support more instance type for custom

* fix: add g4 instances for custom docker model
diff --git a/src/emd/models/custom/custom_docker.py b/src/emd/models/custom/custom_docker.py
@@ -26,6 +26,12 @@
         model_id = CUSTOM_DOCKER_MODEL_ID,
         supported_engines=[custom_engine],
         supported_instances=[
+            g4dnxlarge_instance,
+            g4dn2xlarge_instance,
+            g4dn4xlarge_instance,
+            g4dn8xlarge_instance,
+            g4dn12xlarge_instance,
+            g4dn16xlarge_instance,
             g5dxlarge_instance,
             g5d2xlarge_instance,
             g5d4xlarge_instance,
