fix import

This commit is contained in:
Yu Li
2023-12-19 11:57:45 -06:00
parent 165b1ebdad
commit 1e278887ef

View File

@@ -2,9 +2,9 @@ import sys
import unittest
import torch
sys.path.insert(0, '../airllm')
#sys.path.insert(0, '../airllm')
from auto_model import AutoModel
from airllm.auto_model import AutoModel