fix - AttributeError: 'str' object has no attribute 'removesuffix'

GooeyAI · Sep 5, 2024 · 53d48bc · 53d48bc
1 parent 270ec4c
commit 53d48bc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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: