mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

extensions/openai: Fixes for: embeddings, tokens, better errors. +Docs update, +Images, +logit_bias/logprobs, +more. (#3122)

Browse source
This commit is contained in:
matatonic 2023-07-24 10:28:12 -04:00 • committed by GitHub
parent 1141987a0d
commit 90a4ab631c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 215 additions and 143 deletions
Download patch file Download diff file Expand all files Collapse all files

4
extensions/openai/utils.py
View file

@ -3,9 +3,9 @@ import base64
import numpy as np
def float_list_to_base64(float_list):
def float_list_to_base64(float_array: np.ndarray) -> str:
# Convert the list to a float32 array that the OpenAPI client expects
float_array = np.array(float_list, dtype="float32")
#float_array = np.array(float_list, dtype="float32")
# Get raw bytes
bytes_array = float_array.tobytes()