(self)
| 456 | return samples, intermediates |
| 457 | |
| 458 | def configure_optimizers(self): |
| 459 | lr = self.learning_rate |
| 460 | params = list(self.control_model.parameters()) |
| 461 | if not self.sd_locked: |
| 462 | params += list(self.model.diffusion_model.output_blocks.parameters()) |
| 463 | params += list(self.model.diffusion_model.out.parameters()) |
| 464 | opt = torch.optim.AdamW(params, lr=lr) |
| 465 | return opt |
| 466 | |
| 467 | def low_vram_shift(self, is_diffusing): |
| 468 | if is_diffusing: |
nothing calls this directly
no outgoing calls
no test coverage detected