LLM/airllm
1
0
Fork 0
mirror of https://github.com/0xSojalSec/airllm.git synced 2026-03-07 14:24:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test import

Browse Source
This commit is contained in:
Yu Li
2023-12-19 13:55:57 -06:00
parent 6dcb570e1b
commit 7c13c0e402
3 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
air_llm/__init__.py Normal file
View File

0
air_llm/tests/__init__.py Normal file
View File

5
air_llm/tests/test_automodel.py
View File

@@ -1,10 +1,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