Fixups

Signed-off-by: Ettore Di Giacinto <[email protected]>

Author: mudler

Makefile

@@ -429,6 +429,9 @@ docker-build-kitten-tts:
 docker-save-kitten-tts: backend-images
 	docker save local-ai-backend:kitten-tts -o backend-images/kitten-tts.tar
 
+docker-save-chatterbox: backend-images
+	docker save local-ai-backend:chatterbox -o backend-images/chatterbox.tar
+
 docker-build-kokoro:
 	docker build --build-arg BUILD_TYPE=$(BUILD_TYPE) --build-arg BASE_IMAGE=$(BASE_IMAGE) -t local-ai-backend:kokoro -f backend/Dockerfile.python --build-arg BACKEND=kokoro ./backend
 

backend/python/chatterbox/backend.py

@@ -18,11 +18,19 @@
 import grpc
 
 def is_float(s):
+    """Check if a string can be converted to float."""
     try:
         float(s)
         return True
     except ValueError:
         return False
+def is_int(s):
+    """Check if a string can be converted to int."""
+    try:
+        int(s)
+        return True
+    except ValueError:
+        return False
 
 _ONE_DAY_IN_SECONDS = 60 * 60 * 24
 
@@ -68,10 +76,11 @@ def LoadModel(self, request, context):
             key, value = opt.split(":")
             # if value is a number, convert it to the appropriate type
             if is_float(value):
-                if value.is_integer():
-                    value = int(value)
-                else:
-                    value = float(value)
+                value = float(value)
+            elif is_int(value):
+                value = int(value)
+            elif value.lower() in ["true", "false"]:
+                value = value.lower() == "true"
             self.options[key] = value
 
         self.AudioPath = None