From 893afbd3b37b6f05ca822e0345c2c204a53b915c Mon Sep 17 00:00:00 2001
From: Aiello Chan <aiello.chan@gmail.com>
Date: Tue, 24 Sep 2024 11:17:49 +0800
Subject: [PATCH] fix(2to3): read gpuType from config

Signed-off-by: Aiello Chan <aiello.chan@gmail.com>
---
 src/subCommands/2to3/index.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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'),
         };
       }