Skip to content

Commit

Permalink
Fix import error caused by pydantic on cpu (#11474)
Browse files Browse the repository at this point in the history
  • Loading branch information
plusbang authored Jul 1, 2024
1 parent dbba51f commit 48ad482
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion python/llm/src/ipex_llm/transformers/pipeline_parallel.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 48ad482

Please sign in to comment.