fixed write image error

mudler · Mar 26, 2024 · afa16d3 · afa16d3
1 parent 30685a6
commit afa16d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/python/autogptq/autogptq.py b/backend/python/autogptq/autogptq.py
@@ -5,7 +5,7 @@
 import sys
 import os
 import time
-import urllib.parse
+import base64
 
 import grpc
 import backend_pb2
@@ -114,7 +114,7 @@ def recompile_vl_prompt(self, request):
                 timestamp = str(int(time.time() * 1000))  # Generate timestamp
                 img_path = f"/tmp/vl-{timestamp}.jpg"  # Use timestamp in filename
                 with open(img_path, "wb") as f:
-                    f.write(img)
+                    f.write(base64.b64decode(img))
                 image_paths.append(img_path)
                 prompt = prompt.replace(f"[img-{i}]", "<img>" + img_path + "</img>,")
         else: