Fix import issue in test_vision_openai_server

EvolvingLMMs-Lab · Aug 22, 2024 · dc5f713 · dc5f713
1 parent f7c952e
commit dc5f713
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/srt/test_vision_openai_server.py b/test/srt/test_vision_openai_server.py
@@ -1,8 +1,16 @@
+import base64
+import io
 import json
 import os
+import sys
+import time
 import unittest
 
+import numpy as np
 import openai
+import requests
+from decord import VideoReader, cpu
+from PIL import Image
 
 from sglang.srt.hf_transformers_utils import get_tokenizer
 from sglang.srt.utils import kill_child_process