Merge pull request #144 from weni-ai/fix/update-runtime-cpu

Update: New gpu inputs
weni-ai · Jul 25, 2024 · fc0e4ea · fc0e4ea
2 parents 49208ea + f79fea9
commit fc0e4ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bothub_nlp_api/utils.py b/bothub_nlp_api/utils.py
@@ -188,7 +188,7 @@ def send_job_train_ai_platform(
 
     training_inputs = {
         "scaleTier": "CUSTOM",
-        "masterType": "standard_gpu",
+        "masterType": "standard_p100",
         "masterConfig": {
             "imageUri": f"{settings.BOTHUB_GOOGLE_AI_PLATFORM_REGISTRY}:"
             f"{settings.BOTHUB_GOOGLE_AI_PLATFORM_IMAGE_VERSION}{image_sufix}"