ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Fix llamaindex ut (#10673)

Browse source 
* fix llamaindex ut

* add GPU ut
This commit is contained in:
Zhicun 2024-04-08 09:47:51 +08:00 committed by GitHub
parent a11b708135
commit 321bc69307
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/llm/test/llamaindex/test_llamaindex.py
View file

@ -17,7 +17,7 @@
import pytest import pytest
from unittest import TestCase from unittest import TestCase
import os import os
from ipex_llm.llamaindex.llms import BigdlLLM from ipex_llm.llamaindex.llms import IpexLLM
class Test_LlamaIndex_Transformers_API(TestCase): class Test_LlamaIndex_Transformers_API(TestCase):
def setUp(self): def setUp(self):
@ -50,7 +50,7 @@ class Test_LlamaIndex_Transformers_API(TestCase):
return prompt return prompt
def test_bigdl_llm(self): def test_bigdl_llm(self):
llm = BigdlLLM( llm = IpexLLM.from_model_id(
model_name=self.llama_model_path, model_name=self.llama_model_path,
tokenizer_name=self.llama_model_path, tokenizer_name=self.llama_model_path,
context_window=512, context_window=512,

4
python/llm/test/llamaindex_gpu/test_llamaindex.py
View file

@ -18,7 +18,7 @@ import torch
import pytest import pytest
from unittest import TestCase from unittest import TestCase
import os import os
from ipex_llm.llamaindex.llms import BigdlLLM from ipex_llm.llamaindex.llms import IpexLLM
class Test_LlamaIndex_Transformers_API(TestCase): class Test_LlamaIndex_Transformers_API(TestCase):
def setUp(self): def setUp(self):
@ -51,7 +51,7 @@ class Test_LlamaIndex_Transformers_API(TestCase):
return prompt return prompt
def test_bigdl_llm(self): def test_bigdl_llm(self):
llm = BigdlLLM( llm = IpexLLM.from_model_id(
model_name=self.llama_model_path, model_name=self.llama_model_path,
tokenizer_name=self.llama_model_path, tokenizer_name=self.llama_model_path,
context_window=512, context_window=512,