o HŠ!i?ã@s8ddlmZddlmZddlmZGdd„deƒZdS) é©ÚUnsupportedAPIErroré)Ú BaseTraineré)ÚMODELSc@s2eZdZdZeZdd„Zdd„Zdefdd„Z d S) Ú DocVLMTrainerz&Document Vision Language Model TrainercCstdƒ‚)Nz=Document vision language models do not support train for now.r)ÚselfÚconfig©r úk/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddlex/modules/doc_vlm/trainer.pyÚ__init__sÿzDocVLMTrainer.__init__cCsdS)zupdate training configNr )r r r r Ú update_configszDocVLMTrainer.update_configÚreturncCsd| ¡i}|S)zƒget key-value arguments of model training function Returns: dict: the arguments of training function. Zdevice)Z get_device)r Z train_argsr r r Úget_train_kwargs"s zDocVLMTrainer.get_train_kwargsN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚentitiesr rÚdictrr r r r rs rN)Z utils.errorsrÚbaserZ model_listrrr r r r Ús