diff options
Diffstat (limited to 'train_lora.py')
-rw-r--r-- | train_lora.py | 8 |
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, |