(self, m, sampler, callback=None)
| 5 | |
| 6 | class KDiffusionSampler: |
| 7 | def __init__(self, m, sampler, callback=None): |
| 8 | self.model = m |
| 9 | self.model_wrap = K.external.CompVisDenoiser(m) |
| 10 | self.schedule = sampler |
| 11 | self.generation_callback = callback |
| 12 | |
| 13 | def get_sampler_name(self): |
| 14 | return self.schedule |
nothing calls this directly
no outgoing calls
no test coverage detected