From 19a098e4b99c97e0c8b9257a6be0d2c3e1178028 Mon Sep 17 00:00:00 2001 From: Yu Li Date: Tue, 19 Dec 2023 11:58:35 -0600 Subject: [PATCH] fix import --- air_llm/tests/test_automodel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/air_llm/tests/test_automodel.py b/air_llm/tests/test_automodel.py index 493af23..b546161 100644 --- a/air_llm/tests/test_automodel.py +++ b/air_llm/tests/test_automodel.py @@ -2,7 +2,7 @@ import sys import unittest import torch -#sys.path.insert(0, '../airllm') +#sys.path.insert(0, '..') from airllm.auto_model import AutoModel