Fix images None issue

This commit is contained in:
doctorpangloss 2025-02-15 17:06:27 -08:00
parent 7bf9a86fc1
commit d404ab3185

View File

@ -66,7 +66,8 @@ class OpenAILanguageModelWrapper(LanguageModel):
sampler = sampler or {}
prompt = tokens.get("inputs", [])
prompt = "".join(prompt)
images = tokens.get("images", []) or []
images = tokens.get("images", [])
images = images if images is not None else []
images = [image for image in images if image is not None]
messages: list[ChatCompletionMessageParam] = [
{