From 53d48bc3b25c4621a073cdcbbcbebb869bb49df1 Mon Sep 17 00:00:00 2001
From: Dev Aggarwal <devxpy@gmail.com>
Date: Thu, 5 Sep 2024 14:12:32 +0530
Subject: [PATCH] fix - AttributeError: 'str' object has no attribute
 'removesuffix'

---
 chart/model-values.yaml | 4 ++--
 gooey_gpu.py            | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/chart/model-values.yaml b/chart/model-values.yaml
index a77483d..a32ad6b 100644
--- a/chart/model-values.yaml
+++ b/chart/model-values.yaml
@@ -284,7 +284,7 @@ deployments:
         thenlper/gte-base
 
   - name: "retro-sadtalker"
-    image: "crgooeyprodwestus1.azurecr.io/gooey-gpu-retro:13"
+    image: "crgooeyprodwestus1.azurecr.io/gooey-gpu-retro:14"
     autoscaling:
       queueLength: 2
       minReplicaCount: 3
@@ -299,7 +299,7 @@ deployments:
         SadTalker_V0.0.2_512.safetensors
 
   - name: "retro-wav2lip-gan"
-    image: "crgooeyprodwestus1.azurecr.io/gooey-gpu-retro:13"
+    image: "crgooeyprodwestus1.azurecr.io/gooey-gpu-retro:14"
     autoscaling:
       minReplicaCount: 2
       maxReplicaCount: 10
diff --git a/gooey_gpu.py b/gooey_gpu.py
index 8f5e310..901f888 100644
--- a/gooey_gpu.py
+++ b/gooey_gpu.py
@@ -38,7 +38,7 @@
 )
 
 try:
-    gpu_limit_gib = float(os.environ["RESOURCE_LIMITS_GPU"].removesuffix("Gi"))
+    gpu_limit_gib = float(os.environ["RESOURCE_LIMITS_GPU"].replace("Gi", ""))
 except (KeyError, ValueError):
     print("RESOURCE_LIMITS_GPU environment variable not set to a valid value.")
 else: