Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
oandreeva-nv committed Sep 20, 2024
1 parent e74e86f commit 902429f
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -189,7 +189,9 @@ def setup_lora(self):

# Check if `enable_lora` field is in the `model.json`,
# and if it is, read its contents, which can be string or bool.
if "enable_lora" in self.vllm_engine_config.keys() and str(self.vllm_engine_config["enable_lora"]).lower() == "true":
if (
"enable_lora" in self.vllm_engine_config.keys()
and str(self.vllm_engine_config["enable_lora"]).lower() == "true"
):
# create Triton LoRA weights repository
multi_lora_args_filepath = os.path.join(
Expand Down

0 comments on commit 902429f

Please sign in to comment.