Incompatible with tensorrt_llm==0.10.0

#1
by oobabooga - opened

When I try to load one of these models, I get:

β”‚ /home/user/miniconda3/envs/tensorrt/lib/python3.10/site-packages/tensorrt_llm/runtime/model_runner β”‚
β”‚ .py:85 in _builder_to_model_config                                                                    β”‚
β”‚                                                                                                       β”‚
β”‚    84 def _builder_to_model_config(config: dict) -> Tuple[ModelConfig, dict]:                         β”‚
β”‚ ❱  85     builder_config = config['builder_config']                                                   β”‚
β”‚    86     model_name = builder_config['name']                                                         β”‚
╰───────────────────────────────────────────────────────────────────────────────────────────────────────╯
KeyError: 'builder_config'

This happens because the structure of config.json has changed from 0.9 to 0.10. It would be nice if you could update the config.json files in this repository.

TheFloat16 org

@oobabooga
Hi bro, Thanks for your notice. I'm going to solve this issue this weekend.
Mati

If it will compatible with 0.11.0 it will be perfect!

Sign up or log in to comment