fix(2to3): read gpuType from config

devsapp · Sep 22, 2024 · 40cbad8 · 40cbad8
1 parent 11aca60
commit 40cbad8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/subCommands/2to3/index.ts b/src/subCommands/2to3/index.ts
@@ -269,7 +269,7 @@ export default class SYaml2To3 {
       if (_.get(v.props, 'gpuMemorySize')) {
         v.props.gpuConfig = {
           gpuMemorySize: _.get(v.props, 'gpuMemorySize'),
-          gpuType: 'fc.gpu.tesla.1',
+          gpuType: _.get(v.props, 'instanceType', 'fc.gpu.tesla.1'),
         };
       }