Liam Dodds

Hi! I'm also running in to the exact same issue with the BruteForceSampler. Commenting out the lines you mentioned (which are now [193](https://github.com/optuna/optuna/blob/1925ca1a051b3bf11819478b65d830de614774b2/optuna/samplers/_brute_force.py#L193) and [221](https://github.com/optuna/optuna/blob/1925ca1a051b3bf11819478b65d830de614774b2/optuna/samplers/_brute_force.py#L221) in `samplers/_brute_force.py`) did fix the...

Getting a similar error but with Llama and some `Catcher` class: ```python from awq import AutoAWQForCausalLM from transformers import AutoTokenizer AWQ_CONFIG = { "zero_point": True, "q_group_size": 128, "w_bit": 4, "version":...