summaryrefslogtreecommitdiffstats
path: root/train_lora.py
diff options
context:
space:
mode:
Diffstat (limited to 'train_lora.py')
-rw-r--r--train_lora.py8
1 files changed, 0 insertions, 8 deletions
diff --git a/train_lora.py b/train_lora.py
index fccf48d..b7ee2d6 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -259,12 +259,6 @@ def parse_args():
259 ), 259 ),
260 ) 260 )
261 parser.add_argument( 261 parser.add_argument(
262 "--offset_noise_strength",
263 type=float,
264 default=0,
265 help="Perlin offset noise strength.",
266 )
267 parser.add_argument(
268 "--input_pertubation", 262 "--input_pertubation",
269 type=float, 263 type=float,
270 default=0, 264 default=0,
@@ -1138,7 +1132,6 @@ def main():
1138 sample_output_dir=pti_sample_output_dir, 1132 sample_output_dir=pti_sample_output_dir,
1139 checkpoint_output_dir=pti_checkpoint_output_dir, 1133 checkpoint_output_dir=pti_checkpoint_output_dir,
1140 sample_frequency=pti_sample_frequency, 1134 sample_frequency=pti_sample_frequency,
1141 offset_noise_strength=0,
1142 input_pertubation=args.input_pertubation, 1135 input_pertubation=args.input_pertubation,
1143 no_val=True, 1136 no_val=True,
1144 ) 1137 )
@@ -1291,7 +1284,6 @@ def main():
1291 sample_output_dir=lora_sample_output_dir, 1284 sample_output_dir=lora_sample_output_dir,
1292 checkpoint_output_dir=lora_checkpoint_output_dir, 1285 checkpoint_output_dir=lora_checkpoint_output_dir,
1293 sample_frequency=lora_sample_frequency, 1286 sample_frequency=lora_sample_frequency,
1294 offset_noise_strength=args.offset_noise_strength,
1295 input_pertubation=args.input_pertubation, 1287 input_pertubation=args.input_pertubation,
1296 no_val=args.valid_set_size == 0, 1288 no_val=args.valid_set_size == 0,
1297 avg_loss=avg_loss, 1289 avg_loss=avg_loss,