ChatDrug icon indicating copy to clipboard operation
ChatDrug copied to clipboard

Published 20 hours ago • verified publisher icon chao1224

CUDA OOM Error while importing

Open DingYX0731 opened this issue 6 months ago • 1 comments

It is so weird that CUDA OOM error happens when importing package!!! My GPU is L20 48GB, which should satisfy the requirement of the experiment. Please help... When importing from ChatDrug.task_and_evaluation.Conversational_LLMs_utils import complete and from utils import ( construct_PDDS_prompt, load_retrieval_DB, retrieve_and_feedback, load_thredhold ): image

DingYX0731 avatar Aug 15 '24 04:08 DingYX0731