mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2025-12-06 07:12:10 +01:00
Remove more multimodal extension references
This commit is contained in:
parent
85bf2e15b9
commit
f82667f0b4
|
|
@ -1,13 +1,8 @@
|
||||||
import base64
|
|
||||||
import copy
|
import copy
|
||||||
import re
|
|
||||||
import time
|
import time
|
||||||
from collections import deque
|
from collections import deque
|
||||||
from io import BytesIO
|
|
||||||
|
|
||||||
import requests
|
|
||||||
import tiktoken
|
import tiktoken
|
||||||
from PIL import Image
|
|
||||||
|
|
||||||
from extensions.openai.errors import InvalidRequestError
|
from extensions.openai.errors import InvalidRequestError
|
||||||
from extensions.openai.utils import debug_msg
|
from extensions.openai.utils import debug_msg
|
||||||
|
|
@ -97,28 +92,7 @@ def convert_history(history):
|
||||||
system_message = ""
|
system_message = ""
|
||||||
|
|
||||||
for entry in history:
|
for entry in history:
|
||||||
if "image_url" in entry:
|
content = entry["content"]
|
||||||
image_url = entry['image_url']
|
|
||||||
if "base64" in image_url:
|
|
||||||
image_url = re.sub('^data:image/.+;base64,', '', image_url)
|
|
||||||
img = Image.open(BytesIO(base64.b64decode(image_url)))
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
my_res = requests.get(image_url)
|
|
||||||
img = Image.open(BytesIO(my_res.content))
|
|
||||||
except Exception:
|
|
||||||
raise 'Image cannot be loaded from the URL!'
|
|
||||||
|
|
||||||
buffered = BytesIO()
|
|
||||||
if img.mode in ("RGBA", "P"):
|
|
||||||
img = img.convert("RGB")
|
|
||||||
|
|
||||||
img.save(buffered, format="JPEG")
|
|
||||||
img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
|
|
||||||
content = f'<img src="data:image/jpeg;base64,{img_str}">'
|
|
||||||
else:
|
|
||||||
content = entry["content"]
|
|
||||||
|
|
||||||
role = entry["role"]
|
role = entry["role"]
|
||||||
|
|
||||||
if role == "user":
|
if role == "user":
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue