423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 22:02:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update chat.py

Browse Source 
Former-commit-id: a4f1de9d82743bfd35a6dbc1027242533b22ed5f
This commit is contained in:
marko1616 2024-08-22 12:14:34 +08:00 committed by GitHub
parent e9800212fa
commit 7f6d267e98
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/llamafactory/api/chat.py
View File

@ -17,6 +17,7 @@ import io
import json import json
import os import os
import uuid import uuid
import numpy
from typing import TYPE_CHECKING, AsyncGenerator, Dict, List, Optional, Tuple from typing import TYPE_CHECKING, AsyncGenerator, Dict, List, Optional, Tuple
from ..data import Role as DataRole from ..data import Role as DataRole
@ -112,7 +113,7 @@ def _process_request(
else: # web uri else: # web uri
image_path = requests.get(image_url, stream=True).raw image_path = requests.get(image_url, stream=True).raw
image = Image.open(image_path).convert("RGB") image = numpy.array(Image.open(image_path).convert("RGB"))
else: else:
input_messages.append({"role": ROLE_MAPPING[message.role], "content": message.content}) input_messages.append({"role": ROLE_MAPPING[message.role], "content": message.content})