(self, m, sampler)
| 605 | |
| 606 | class KDiffusionSampler: |
| 607 | def __init__(self, m, sampler): |
| 608 | self.model = m |
| 609 | self.model_wrap = K.external.CompVisDenoiser(m) |
| 610 | self.schedule = sampler |
| 611 | |
| 612 | def get_sampler_name(self): |
| 613 | return self.schedule |
nothing calls this directly
no outgoing calls
no test coverage detected