mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 17:11:21 +08:00
fix paddleformers fallback (#6465)
This commit is contained in:
@@ -194,7 +194,10 @@ class DataProcessor(BaseDataProcessor):
|
||||
eos_token is {self.tokenizer.eos_token}, {self.tokenizer.eos_token_id} "
|
||||
)
|
||||
|
||||
from paddleformers.trl.llm_utils import get_eos_token_id
|
||||
try:
|
||||
from paddleformers.trl.llm_utils import get_eos_token_id
|
||||
except Exception:
|
||||
from paddleformers.cli.utils.llm_utils import get_eos_token_id
|
||||
|
||||
self.eos_token_ids = get_eos_token_id(self.tokenizer, self.generation_config)
|
||||
data_processor_logger.info(
|
||||
|
||||
Reference in New Issue
Block a user