remove unneeded code.

sayakpaul · sayakpaul · commit db8bbbda5b64 · 2023-04-05T15:01:55.000+05:30
diff --git a/examples/text_to_image/train_text_to_image.py b/examples/text_to_image/train_text_to_image.py
@@ -564,9 +564,6 @@ def load_model_hook(models, input_dir):
             args.learning_rate * args.gradient_accumulation_steps * args.train_batch_size * accelerator.num_processes
         )
 
-    # if args.snr_gamma is not None:
-    #     snr_fn = compute_snr(noise_scheduler)
-
     # Initialize the optimizer
     if args.use_8bit_adam:
         try:

Original file line number	Diff line number	Diff line change
`@@ -564,9 +564,6 @@ def load_model_hook(models, input_dir):`
`564`	`564`	`args.learning_rate * args.gradient_accumulation_steps * args.train_batch_size * accelerator.num_processes`
`565`	`565`	`)`
`566`	`566`
`567`		`- # if args.snr_gamma is not None:`
`568`		`- # snr_fn = compute_snr(noise_scheduler)`
`569`		`-`
`570`	`567`	`# Initialize the optimizer`
`571`	`568`	`if args.use_8bit_adam:`
`572`	`569`	`try:`