Starting LoRAX with mistralai/Mixtral-8x7B-Instruct-v0.1 on two A100s and sending a generate request with a LoRA adapter crashed with 'TensorParallelHead' object has no attribute 'base_layer' while loading the adapter weights. Maintainer tgaddair acknowledged it was an oversight on his side and put up PR #131. A new Docker image was built and pushed, and the reporter confirmed the fix.