Revert removing the unused imports (#385)

2024-04-23 22:36:33 +08:00
parent 9acc6e3504
commit 150d7020ed
33 changed files with 72 additions and 26 deletions
--- a/test/srt/model/test_llama_extend.py
+++ b/test/srt/model/test_llama_extend.py
@@ -1,6 +1,10 @@
 import multiprocessing
 import os
+import time

+import numpy as np
+import torch
+import torch.distributed as dist
 import transformers

 from sglang.srt.managers.router.infer_batch import Batch, ForwardMode, Req