From 2012336426d167750e7c0880b9052bb9abf04cc2 Mon Sep 17 00:00:00 2001 From: Ruheena Suhani Shaik Date: Thu, 2 Jan 2025 14:42:01 +0530 Subject: [PATCH] Lora manager tests fix (#652) This PR solves the "ModuleNotFoundError: No Module named torch.hpu" in test_lora_manager_hpu.py::test_from_lora_tensors by importing "habana_frameworks.hpu" to lora model. Co-authored-by: Vivek Goel --- tests/lora/test_lora_manager_hpu.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/lora/test_lora_manager_hpu.py b/tests/lora/test_lora_manager_hpu.py index cca4802c4cd84..ede3b11e431f5 100644 --- a/tests/lora/test_lora_manager_hpu.py +++ b/tests/lora/test_lora_manager_hpu.py @@ -1,6 +1,7 @@ import os from typing import Dict, List +import habana_frameworks.torch # noqa: F401 import pytest import torch from safetensors.torch import load_file