Add RestoreDPMPP2MSampler

Better suited for lightning models
2024-03-10 14:25:50 +02:00
parent e0324bed08
commit c1e8bc5f80
3 changed files with 179 additions and 4 deletions
@@ -105,6 +105,7 @@ class SUPIRModel(DiffusionEngine):
        self.sampler_config.params.s_noise = s_noise
        self.sampler = instantiate_from_config(self.sampler_config)

+        print("Sampler: ", self.sampler_config.target)
        print("sampler_config: ", self.sampler_config.params)

        if seed == -1: