Skip to content

Commit e74e86f

Browse files
oandreeva-nvkthui
andauthored
Update src/model.py
Co-authored-by: Jacky <[email protected]>
1 parent e03fabc commit e74e86f

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

src/model.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -189,17 +189,7 @@ def setup_lora(self):
189189

190190
# Check if `enable_lora` field is in the `model.json`,
191191
# and if it is, read its contents, which can be string or bool.
192-
if "enable_lora" in self.vllm_engine_config.keys() and (
193-
(
194-
isinstance(self.vllm_engine_config["enable_lora"], str)
195-
and self.vllm_engine_config["enable_lora"].lower() == "true"
196-
)
197-
or (
198-
(
199-
isinstance(self.vllm_engine_config["enable_lora"], bool)
200-
and self.vllm_engine_config["enable_lora"]
201-
)
202-
)
192+
if "enable_lora" in self.vllm_engine_config.keys() and str(self.vllm_engine_config["enable_lora"]).lower() == "true":
203193
):
204194
# create Triton LoRA weights repository
205195
multi_lora_args_filepath = os.path.join(

0 commit comments

Comments
 (0)