From 500e974f5eefba94cdb4497b57ddda149f59fc97 Mon Sep 17 00:00:00 2001 From: mayukhdeb Date: Sun, 17 Mar 2024 00:49:01 -0700 Subject: [PATCH] remove unused arg (things are working) --- trainer_pti.py | 1 - 1 file changed, 1 deletion(-) diff --git a/trainer_pti.py b/trainer_pti.py index 03f8e63..c78ff54 100755 --- a/trainer_pti.py +++ b/trainer_pti.py @@ -283,7 +283,6 @@ def main( gradient_accumulation_steps: int = 1, # todo unet_learning_rate: float = 1.0, ti_lr: float = 3e-4, - lora_lr: float = 1.0, prodigy_d_coef: float = 0.33, l1_penalty: float = 0.0, lora_weight_decay: float = 0.005,