fix divisor

Author: ethansmith2000

examples/dreambooth/train_dreambooth.py

@@ -1303,10 +1303,12 @@ def compute_text_embeddings(prompt):
 
                     if noise_scheduler.config.prediction_type == "v_prediction":
                         # Velocity objective needs to be floored to an SNR weight of one.
-                        snr = snr + 1             
+                        divisor = snr + 1         
+                    else:
+                        divisor = snr
 
                     mse_loss_weights = (
-                        torch.stack([snr, args.snr_gamma * torch.ones_like(timesteps)], dim=1).min(dim=1)[0] / snr
+                        torch.stack([snr, args.snr_gamma * torch.ones_like(timesteps)], dim=1).min(dim=1)[0] / divisor
                     )
 
                     loss = F.mse_loss(model_pred.float(), target.float(), reduction="none")