added htorch support to enable hpu device

HabanaAI · Dec 23, 2024 · 157cdee · 157cdee
1 parent 88ef381
commit 157cdee
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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