From bf892826f767b275e46aa2bfda867693ce8a223a Mon Sep 17 00:00:00 2001 From: mayukhdeb Date: Fri, 15 Mar 2024 00:48:47 -0700 Subject: [PATCH] oopsie --- trainer/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/trainer/trainer.py b/trainer/trainer.py index abca634..5cf71fb 100644 --- a/trainer/trainer.py +++ b/trainer/trainer.py @@ -49,8 +49,8 @@ class Trainer: def train(self): - if args.concept_mode == "style": # for styles you usually want the LoRA matrices to absorb a lot (instead of just the token embedding) - args.l1_penalty = 0.05 + if self.args.concept_mode == "style": # for styles you usually want the LoRA matrices to absorb a lot (instead of just the token embedding) + self.args.l1_penalty = 0.05 args = self.args