[Bugfix][CI/Build][Hardware][AMD] Install matching torchvision to fix AMD tests (#5949)

This commit is contained in:
Matt Wong
2024-06-29 14:47:58 -05:00
committed by GitHub
parent 75aa1442db
commit 9def10664e
2 changed files with 14 additions and 8 deletions

View File

@@ -14,7 +14,7 @@ import torch
from huggingface_hub import snapshot_download
from openai import BadRequestError
from ..utils import VLLM_PATH, RemoteOpenAIServer
from ..utils import RemoteOpenAIServer
# any model with a chat template should work here
MODEL_NAME = "HuggingFaceH4/zephyr-7b-beta"
@@ -79,7 +79,7 @@ def zephyr_lora_files():
@pytest.fixture(scope="module")
def ray_ctx():
ray.init(runtime_env={"working_dir": VLLM_PATH})
ray.init()
yield
ray.shutdown()