fix(huggingface_restapi.py): add support for additional hf embedding formats

2023-12-15 21:02:41 -08:00 · 2023-12-15 21:02:41 -08:00 · add153d110
commit add153d110
parent 5fe5149070
2 changed files with 9 additions and 2 deletions
--- a/litellm/llms/huggingface_restapi.py
+++ b/litellm/llms/huggingface_restapi.py
@ -584,6 +584,14 @@ class Huggingface(BaseLLM):
                            "embedding": embedding # flatten list returned from hf
                        }
                    )
+                elif isinstance(embedding, list) and isinstance(embedding[0], float):
+                    output_data.append(
+                        {
+                            "object": "embedding",
+                            "index": idx,
+                            "embedding": embedding # flatten list returned from hf
+                        }
+                    )
                else: 
                    output_data.append(
                        {