handle query params better in civitai client

ssube · Dec 10, 2023 · e91e084 · e91e084
1 parent e200fe9
commit e91e084
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/api/onnx_web/convert/client/civitai.py b/api/onnx_web/convert/client/civitai.py
@@ -56,6 +56,9 @@ def download(
 
         if self.token:
             logger.debug("adding Civitai token authentication")
-            source = f"{source}?token={self.token}"
+            if "?" in source:
+                source = f"{source}&token={self.token}"
+            else:
+                source = f"{source}?token={self.token}"
 
         return download_progress(source, cache_paths[0])