diff --git a/python/llm/src/ipex_llm/transformers/pipeline_parallel.py b/python/llm/src/ipex_llm/transformers/pipeline_parallel.py index 6db55d53..07615449 100644 --- a/python/llm/src/ipex_llm/transformers/pipeline_parallel.py +++ b/python/llm/src/ipex_llm/transformers/pipeline_parallel.py @@ -32,7 +32,11 @@ logger = logging.getLogger(__name__) import asyncio import uuid import threading -from pydantic import BaseModel +try: + from pydantic import BaseModel +except ImportError: + from abc import ABCMeta + BaseModel = ABCMeta # patch GenerationMixin.generate from transformers import GenerationMixin