Update serve/model_worker.py

serve/model_worker.py

@@ -15,12 +15,12 @@ from functools import partial
 from transformers import TextIteratorStreamer
 from threading import Thread
 
-from .constants import WORKER_HEART_BEAT_INTERVAL
-from .utils import (build_logger, server_error_msg, pretty_print_semaphore)
-from .builder import load_pretrained_model
-from .mm_utils import process_images, load_image_from_base64, tokenizer_image_token, get_model_name_from_path, \
+from serve.constants import WORKER_HEART_BEAT_INTERVAL
+from serve.utils import (build_logger, server_error_msg, pretty_print_semaphore)
+from serve.builder import load_pretrained_model
+from serve.mm_utils import process_images, load_image_from_base64, tokenizer_image_token, get_model_name_from_path, \
     KeywordsStoppingCriteria
-from .constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
+from serve.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
 
 GB = 1 << 30