diff --git a/src/llamafactory/api/chat.py b/src/llamafactory/api/chat.py index 1010aab1..5bad1af2 100644 --- a/src/llamafactory/api/chat.py +++ b/src/llamafactory/api/chat.py @@ -17,7 +17,7 @@ import io import json import os import uuid -import numpy +import numpy as np from typing import TYPE_CHECKING, AsyncGenerator, Dict, List, Optional, Tuple from ..data import Role as DataRole @@ -113,7 +113,7 @@ def _process_request( else: # web uri image_path = requests.get(image_url, stream=True).raw - image = numpy.array(Image.open(image_path).convert("RGB")) + image = np.array(Image.open(image_path).convert("RGB")) else: input_messages.append({"role": ROLE_MAPPING[message.role], "content": message.content})