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

Fix import error caused by pydantic on cpu (#11474)

Browse source
This commit is contained in:
binbin Deng 2024-07-01 15:49:49 +08:00 committed by GitHub
parent dbba51f455
commit 48ad482d3d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/llm/src/ipex_llm/transformers/pipeline_parallel.py
View file

@ -32,7 +32,11 @@ logger = logging.getLogger(__name__)
import asyncio import asyncio
import uuid import uuid
import threading import threading
from pydantic import BaseModel try:
from pydantic import BaseModel
except ImportError:
from abc import ABCMeta
BaseModel = ABCMeta
# patch GenerationMixin.generate # patch GenerationMixin.generate
from transformers import GenerationMixin from transformers import GenerationMixin