remove unused arg (things are working)

This commit is contained in:
mayukhdeb
2024-03-17 00:49:01 -07:00
parent 4673ec5ef4
commit 500e974f5e
-1
View File
@@ -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,