format

Signed-off-by: shen-shanshan <[email protected]>

Author: shen-shanshan

tests/conftest.py

@@ -18,11 +18,11 @@
 #
 
 import contextlib
-import gc
 from typing import List, Optional, Tuple, TypeVar, Union
 
 import numpy as np
 import pytest
+import torch
 from huggingface_hub import snapshot_download
 from PIL import Image
 from vllm import LLM, SamplingParams