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

RyuKosei · Jul 19, 2024 · f239018 · f239018
1 parent ea2472b
commit f239018
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/llm/src/ipex_llm/transformers/pipeline_parallel.py b/python/llm/src/ipex_llm/transformers/pipeline_parallel.py
@@ -32,7 +32,11 @@
 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