Skip to content

Commit

Permalink
Added the changes to conftest to fix test_lora_manager
Browse files Browse the repository at this point in the history
  • Loading branch information
rsshaik1 committed Sep 30, 2024
1 parent 188bd3a commit f59495a
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion tests/lora/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -262,7 +262,6 @@ def get_model_patched(*, model_config, device_config, **kwargs):
return get_model_old(model_config=model_config,
device_config=device_config,
**kwargs)

if is_hpu():
with patch("vllm.worker.habana_model_runner.get_model", get_model_patched):
engine = vllm.LLM("meta-llama/Llama-2-7b-hf", enable_lora=False)
Expand Down

0 comments on commit f59495a

Please sign in to comment.