Update examples/dreambooth/train_dreambooth_lora_sana.py

sayakpaul · a-r-r-o-w · web-flow · commit 31b1a8ed593d · 2024-12-23T11:59:45.000+05:30
Co-authored-by: Aryan &lt;aryan@huggingface.co&gt;
diff --git a/examples/dreambooth/train_dreambooth_lora_sana.py b/examples/dreambooth/train_dreambooth_lora_sana.py
@@ -1370,7 +1370,6 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
                 noisy_model_input = (1.0 - sigmas) * model_input + sigmas * noise
 
                 # Predict the noise residual
-                print(f"{noisy_model_input.dtype=}, {prompt_embeds.dtype=}")
                 model_pred = transformer(
                     hidden_states=noisy_model_input,
                     encoder_hidden_states=prompt_embeds,
