train()

Author: sayakpaul

examples/control-lora/train_control_lora_flux.py

@@ -1139,6 +1139,7 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
 
     image_logs = None
     for epoch in range(first_epoch, args.num_train_epochs):
+        flux_transformer.train()
         for step, batch in enumerate(train_dataloader):
             with accelerator.accumulate(flux_transformer):
                 # Convert images to latent space