diff --git a/src/subCommands/2to3/index.ts b/src/subCommands/2to3/index.ts index 09e8825..bb9c83a 100644 --- 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'), }; }