o }o™iã@s>ddlZddlmZddlmZdZdZdedefdd „Z dS) éN)ÚPath)ÚTarPathz config.jsonzrank{}.safetensorsÚpathÚreturncCsDtj |¡r t|ƒ}nt|ƒ}|t}|t d¡}| ¡o!| ¡S)zjDetect if a given path is a TensorRT-LLM a.k.a. "qnemo" checkpoint based on config & tensor data presence.r) ÚosrÚisdirrrÚ CONFIG_NAMEÚ WEIGHTS_NAMEÚformatÚexists)rÚ config_pathÚ tensor_path©rúS/home/ubuntu/.local/lib/python3.10/site-packages/nemo/export/trt_llm/qnemo/utils.pyÚis_qnemo_checkpoints  r) rÚpathlibrÚnemo.export.tarutilsrrr ÚstrÚboolrrrrrÚs