From beddf0152a06e04082b5445695393171eb95a849 Mon Sep 17 00:00:00 2001 From: harikdava Date: Thu, 23 Jan 2025 10:20:38 +0100 Subject: [PATCH] extra lora args added --- predict.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/predict.py b/predict.py index 3a8f0a1..4ff40c3 100644 --- a/predict.py +++ b/predict.py @@ -949,8 +949,10 @@ def predict( lora_weights: str = Inputs.lora_weights, lora_scale: float = Inputs.lora_scale, megapixels: str = Inputs.megapixels, + extra_lora_weights: str = Inputs.lora_weights, + extra_lora_scale: float = Inputs.lora_scale, ) -> List[Path]: - self.handle_loras(go_fast, lora_weights, lora_scale) + self.handle_loras(go_fast, lora_weights, lora_scale, extra_lora_weights, extra_lora_scale) width, height = self.size_from_aspect_megapixels(aspect_ratio, megapixels) imgs, np_imgs = self.shared_predict( @@ -1004,12 +1006,14 @@ def predict( lora_weights: str = Inputs.lora_weights, lora_scale: float = Inputs.lora_scale, megapixels: str = Inputs.megapixels, + extra_lora_weights: str = Inputs.lora_weights, + extra_lora_scale: float = Inputs.lora_scale, ) -> List[Path]: if image and go_fast: print("img2img not supported with fp8 quantization; running with bf16") go_fast = False - self.handle_loras(go_fast, lora_weights, lora_scale) + self.handle_loras(go_fast, lora_weights, lora_scale, extra_lora_weights, extra_lora_scale) width, height = self.size_from_aspect_megapixels(aspect_ratio, megapixels) imgs, np_imgs = self.shared_predict(